oskarastrom commited on
Commit
c1b62e3
·
1 Parent(s): 84bfeb1

Update detect_frames.py

Browse files
Files changed (1) hide show
  1. 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().numpy())
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)