Torchscript file is not working with detect.py

[RizwanMunawar]

Search before asking

• [X] I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Detection

Bug

Below, command is throwing error, when I am detecting with torchscript file. python .\detect.py --weights "B:\yolov5-master\yolov5s.torchscript" --source "B:\Testing videos\test.avi"

,

Environment

Windows 10 Python 3.7 yolov5 latest release

Minimal Reproducible Example

python .\detect.py --weights "B:\yolov5-master\yolov5s.torchscript" --source "B:\Testing videos\test.avi"

Additional

No response

Are you willing to submit a PR?

• [X] Yes I'd like to help by submitting a PR!

[glenn-jocher]

@RizwanMunawar thanks for the bug report! I am able to reproduce this problem. I'll add a TODO to investigate today.

!python export.py --weights yolov5s.pt --include torchscript
!python detect.py --weights yolov5s.torchscript

[glenn-jocher]

@RizwanMunawar good news 😃! Your original issue may now be fixed ✅ in PR #6007. To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

[RizwanMunawar]

Cool, it works fine now.