YOLOv5-Lite icon indicating copy to clipboard operation
YOLOv5-Lite copied to clipboard

Published 20 hours ago • verified publisher icon ppogg

Unable to run detect.py for testing

Open joel5638 opened this issue 2 years ago • 2 comments

Im testing on few images. Im unable to run python3 detect.py --source home/joel/img4.jpg --weights v5lite-s.pt.

Im getting this error: AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

Tried resolving the environmental issues too Nothing is working. I have created new env and install requirements.txt. Still did not work. I have also used --device as cpu. Still nothing works

joel5638 avatar Jul 14 '22 05:07 joel5638

Thanks for using YOLOv5 Lite！In fact, this is caused by an overly high version of pytorch. Please follow the requirements.txt to install BTW，the torch version in my env is 1.8.0.

ppogg avatar Jul 15 '22 05:07 ppogg

Thank you for this amazing project. Will you support the newest version of pytorch?

gamegrd avatar Sep 15 '22 10:09 gamegrd