Spaces:
Runtime error
Runtime error
Commit
·
ed9fba4
1
Parent(s):
409a900
Update detect_frames.py
Browse files- 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()
|