littlelight-e
commented
2 years ago 这个错误只要把OpenCV版本切换到4.2即可解决，但是又遇到了新问题如下，猜想会不会是因为torch的版本问题，我想将torch版本切换到1.6可是，提示我不能切换。
image 1/1 D:\Weld inspection\YOLOv5\rotation-yolov5-master\inference\images\1.tif: D:\Weld inspection\YOLOv5\rotation-yolov5-master\venv\lib\site-packages\torch\nn\functional.py:718: UserWarning: Named tensors and all their associated APIs are an experimental feature and subject to change. Please do not use them for anything important until they are released as stable. (Triggered internally at ..\c10/core/TensorImpl.h:1156.)
return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)
Traceback (most recent call last):
File "D:/Weld inspection/YOLOv5/rotation-yolov5-master/detect.py", line 174, in
其余的应该要自己训练，
如果想使用gpu，我猜，不同于在YOLOv5.v5中，只要安装适合自己电脑的cuda和cudnn，再安装pytorch框架即可，而要安装作者指定的版本。
huyunlei
Fusing layers... Model Summary: 188 layers, 2.22122e+07 parameters, 0 gradients image 1/1 D:\Weld inspection\YOLOv5\rotation-yolov5-master\inference\images\1.tif: Traceback (most recent call last): File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\detect.py", line 174, in
detect()
File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\detect.py", line 79, in detect
pred = rotate_non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=True)
File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\utils\general.py", line 869, in rotate_non_max_suppression
i = nms(boxes, scores)
File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\utils\general.py", line 797, in nms
rotate_ious = rotate_iou(max_iou_boxes[-1], sort_boxes[1:])
File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\utils\general.py", line 710, in rotate_iou
r_b1 = get_rotated_coors(box1[i])
File "D:\Weld inspection\YOLOv5\rotation-yolov5-master\utils\general.py", line 729, in get_rotated_coors
R[:2] = cv2.getRotationMatrix2D(angle=-a*180/math.pi, center=(cx,cy), scale=1)
TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type