RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 3 3, but got 3-dimensional in

使用Pytorch运行时,出现下述错误:RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 3 3, but got 3-dimensional input of size [3, 224, 224] instead

解决方案:

    img=Image.open(imgpath)        # 读取图片
    img=img.resize((TARGET_IMG_SIZE, TARGET_IMG_SIZE))
    tensor=img_to_tensor(img)    # 将图片转化成tensor,

    print(tensor.shape)  #[3, 224, 224]
    tensor = Variable(torch.unsqueeze(tensor, dim=0).float(), requires_grad=False)

    print(tensor.shape)  #[1,3, 224, 224]
    tensor=tensor.cuda()

 

 补充:

    from torchvision import models, transforms

     VGG = models.vgg16(pretrained=True)
     feature = torch.nn.Sequential(*list(VGG.children())[:])
     print(feature)
     print('=============')
     print(VGG._modules.keys())# 查看包含哪两部分
     exit()

  • 7
    点赞
  • 14
    评论
  • 23
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值