Update paths of torch models

2023-01-19 22:10:34 +01:00 · 2023-01-19 22:10:34 +01:00 · 93f3296eff
commit 93f3296eff
parent f9b2c25fe5
1 changed files with 2 additions and 4 deletions
--- a/jetson-deployment/model.py
+++ b/jetson-deployment/model.py
@ -158,8 +158,6 @@ if __name__ == '__main__':
    opt = parser.parse_args()

    if opt.source:
-        detect(opt.source, 'runs/train/yolov7-custom7/weights/best.pt',
-               'resnet.pt')
+        detect(opt.source, 'yolo.pt', 'resnet.pt')
    if opt.onnx:
-        export_to_onnx('runs/train/yolov7-custom7/weights/best.pt',
-                       'resnet.pt')
+        export_to_onnx('yolo.pt', 'resnet.pt')