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])
|