sindhuhegde commited on
Commit
bd49b58
·
1 Parent(s): a6d3a9f

Update app

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. preprocess/inference_preprocess.py +1 -1
app.py CHANGED
@@ -1090,7 +1090,7 @@ def process_video_activespeaker(video_path, global_speaker, num_avg_frames):
1090
  return None, status
1091
 
1092
  # Pre-process and extract per-speaker tracks in each scene
1093
- # print("Pre-processing the input video...")
1094
  # status = subprocess.call("python preprocess/inference_preprocess.py --data_dir={}/temp --sd_root={}/crops --work_root={}/metadata --data_root={}".format(result_folder_input, result_folder_input, result_folder_input, video_path), shell=True)
1095
  # if status != 0:
1096
  # msg = "Error in pre-processing the input video, please check the input video and try again..."
 
1090
  return None, status
1091
 
1092
  # Pre-process and extract per-speaker tracks in each scene
1093
+ print("Pre-processing the input video...")
1094
  # status = subprocess.call("python preprocess/inference_preprocess.py --data_dir={}/temp --sd_root={}/crops --work_root={}/metadata --data_root={}".format(result_folder_input, result_folder_input, result_folder_input, video_path), shell=True)
1095
  # if status != 0:
1096
  # msg = "Error in pre-processing the input video, please check the input video and try again..."
preprocess/inference_preprocess.py CHANGED
@@ -33,7 +33,7 @@ parser.add_argument('--work_root', type=str, required=True, help='Path to save m
33
  parser.add_argument('--data_root', type=str, required=True, help='Directory containing ONLY full uncropped videos')
34
  opt = parser.parse_args()
35
 
36
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
37
 
38
  def bb_intersection_over_union(boxA, boxB):
39
  xA = max(boxA[0], boxB[0])
 
33
  parser.add_argument('--data_root', type=str, required=True, help='Directory containing ONLY full uncropped videos')
34
  opt = parser.parse_args()
35
 
36
+ # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
37
 
38
  def bb_intersection_over_union(boxA, boxB):
39
  xA = max(boxA[0], boxB[0])