pytorch训练的维度不匹配的问题，解决办法

扩展单张图片维度

代码语言：javascript复制

import cv2
import torch

image = cv2.imread(img_path)
image = torch.tensor(image)
print(image.size())

img = image.view(1,

pytorch

0 人点赞