如何利用opencv读取图片并且转换成pytorch可以使用的形式？

如何利用opencv读取图片并且转换成pytorch可以使用的形式？

1.首先需要注意的是opencv读取图片的格式以及维度，后续转换需要注意

opencv读取图片后的维度是B,H,W,C 并且读取的是bgr

2.如何利用opencv读取图片并且转换成pytorch可以使用的形式：

img = cv2.imread(img_dir)
resized_img = cv2.resize(img, new_size, interpolation=cv2.INTER_LINEAR)#
resized_img_rgb = cv2.COLOR_BGR2RGB(resized_img)
resized_img_rgb = cv2.cvtColor(resized_img, cv2.COLOR_BGR2RGB)
imgout = torch.from_numpy(resized_img_rgb).to("cuda").div(255.0).unsqueeze(0)
imgout2 = imgout.permute(0, 3, 1, 2)
output = model(imgout2.float())