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

2022-10-05 12:07:55

扩展单张图片维度

import cv2import torch

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

img= image.view(1,
  • 作者：三更两点
  • 原文链接：https://errol.blog.csdn.net/article/details/113928972
    更新时间：2022-10-05 12:07:55