Spaces:
Configuration error
Configuration error
Commit
·
bd49b58
1
Parent(s):
a6d3a9f
Update app
Browse files- app.py +1 -1
- 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 |
-
|
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])
|