在封装yolov5-7.0的detect文件时遇到这个问题
封装代码参考:魔改并封装 YoloV5 Version7 的 detect.py 成 API接口以供 python 程序使用_yolov5部署成api接口-CSDN博客
这个错误表明输入类型(float)和偏置类型(struct c10::Half)不匹配
由于这位博主使用的是cpu进行检测,所以不会遇到半精度的问题
当我们使用gpu进行检测时,只需要这样
im = im.half() if self.model.fp16 else im.float() # uint8 to fp16/32
# 改为
im = im.half()
# 也就是注释掉后面的判断
这样就成功解决啦。