yolov5 detect.py报错

yolov5 detect出现:
RuntimeError: The size of tensor a (38) must match the size of tensor b (49) at non-singleton dimension 3
原因:yolov5版本不一致(训练时用较高版本,检测用较低版本)
解决办法:采用yolov5相同版本检测

你可能感兴趣的:(深度学习,pytorch,python)