zenon/master-thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update paths of torch models

Browse Source
This commit is contained in:
Tobias Eidelpes 2023-01-19 22:10:34 +01:00
parent f9b2c25fe5
commit 93f3296eff
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
jetson-deployment/model.py
View File

@ -158,8 +158,6 @@ if __name__ == '__main__':
opt = parser.parse_args() opt = parser.parse_args()
if opt.source: if opt.source:
detect(opt.source, 'runs/train/yolov7-custom7/weights/best.pt', detect(opt.source, 'yolo.pt', 'resnet.pt')
'resnet.pt')
if opt.onnx: if opt.onnx:
export_to_onnx('runs/train/yolov7-custom7/weights/best.pt', export_to_onnx('yolo.pt', 'resnet.pt')
'resnet.pt')