oskarastrom commited on
Commit
ed9fba4
·
1 Parent(s): 409a900

Update detect_frames.py

Browse files
Files changed (1) hide show
  1. scripts/detect_frames.py +1 -1
scripts/detect_frames.py CHANGED
@@ -78,7 +78,7 @@ def detect(in_seq_dir, out_seq_dir, model, device, verbose):
78
  with open(os.path.join(out_seq_dir, 'pred.json'), 'w') as f:
79
  json.dump(json_obj, f)
80
 
81
- torch.save(inference, 'inference.pt')
82
 
83
  def argument_parser():
84
  parser = argparse.ArgumentParser()
 
78
  with open(os.path.join(out_seq_dir, 'pred.json'), 'w') as f:
79
  json.dump(json_obj, f)
80
 
81
+ torch.save(inference, os.path.join(out_seq_dir, 'inference.pt'))
82
 
83
  def argument_parser():
84
  parser = argparse.ArgumentParser()