Spaces:
Runtime error
Runtime error
Commit
·
c1b62e3
1
Parent(s):
84bfeb1
Update detect_frames.py
Browse files- scripts/detect_frames.py +1 -1
scripts/detect_frames.py
CHANGED
@@ -73,7 +73,7 @@ def detect(in_seq_dir, out_seq_dir, model, device, verbose):
|
|
73 |
with tqdm(total=len(inference), desc="...") as pbar:
|
74 |
for tensor in inference:
|
75 |
if isinstance(tensor, torch.Tensor):
|
76 |
-
inference_list.append(tensor.cpu().
|
77 |
else:
|
78 |
inference_list.append(tensor)
|
79 |
pbar.update(1)
|
|
|
73 |
with tqdm(total=len(inference), desc="...") as pbar:
|
74 |
for tensor in inference:
|
75 |
if isinstance(tensor, torch.Tensor):
|
76 |
+
inference_list.append(tensor.cpu().tolist())
|
77 |
else:
|
78 |
inference_list.append(tensor)
|
79 |
pbar.update(1)
|