訓練yolov5的時候如果啟用了Albumentations庫可能會出現(xiàn)：

    cv2.cvtColor(im_hsv, cv2.COLOR_HSV2BGR, dst=im)  # no return needed
TypeError: Expected Ptr<cv::UMat> for argument 'dst'

或者

    cv2.cvtColor(im_hsv, cv2.COLOR_HSV2BGR, dst=im)  # no return needed
cv2.error: OpenCV(4.6.0) :-1: error: (-5:Bad argument) in function 'cvtColor'
> Overload resolution failed:
>  - Layout of the output array dst is incompatible with cv::Mat
>  - Expected Ptr<cv::UMat> for argument 'dst'

原因是之前的操作使得im在內(nèi)存中不連續(xù)，在augment_hsv之前加一行解決：

img = np.ascontiguousarray(img)