调用检测的接口报错‘RuntimeError: "unfolded2d_copy" not implemented for 'Half'，请问在哪里修改use_half=False呢？

Sharpiless / Yolov5-Flask-VUE

基于Flask+VUE前后端，在阿里云公网WEB端部署YOLOv5目标检测模型

MIT License

183 stars 24 forks source link

Open xd905 opened 3 years ago

Sharpiless commented 3 years ago

这是CUDA没有安装的原因

Sharpiless commented 3 years ago

注意目前版本不支持cpu运行

dddlli commented 3 years ago

将detect中的half改为float就可以了

xienova commented 7 months ago

将detect中的half改为float就可以了

将AIDetector_pytorch.py中的两个half()换为float()

hh0611cc0708 commented 6 months ago

将detect中的half改为float就可以了

将AIDetector_pytorch.py中的两个half()换为float()

感谢回答