_泡菜。不吃腌菜:取消拾取堆栈und

2024-05-15 03:55:25 发布

您现在位置:Python中文网/ 问答频道 /正文

我是一个新的深度学习者,我试图重复https://github.com/swamiviv/LSD-seg中的代码,这是用于图像语义分割的,但在Pytorch中发生了一个问题,我真的需要你的帮助。在

非常感谢。在

以下是环境:

Ubuntu 14.04.1
Python 3.6.4
Pytorch 0.4.0

代码是用来下载预先训练好的模型的,但是我的服务器无法访问googledrive,所以我从其他方法下载了模型,并将模型文件更新到服务器上,并修改了相关代码,但是没有成功。在

原代码:

^{pr2}$

我用其他方法下载了模型文件放到特定位置后,修改了代码:

import os.path as osp
import fcn
import torch
import torchvision

def VGG16(pretrained=False):
    model = torchvision.models.vgg16(pretrained=False)
    if not pretrained:
        return model

    state_dict = torch.load('/home/sunyuze/LSD/data/vgg16_from_caffe.pth')
    model.load_state_dict(state_dict)
    return model

错误警报为:

Traceback (most recent call last):
  File "train.py", line 157, in <module>
    main()
  File "train.py", line 76, in main
    vgg16 = torchfcn.models.VGG16(pretrained=True)
  File "/home/sunyuze/LSD/code/torchfcn/models/vgg.py", line 17, in VGG16
    state_dict = torch.load('/home/sunyuze/LSD/data/vgg16_from_caffe.pth')
  File "/home/sunyuze/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
    return _load(f, map_location, pickle_module)
  File "/home/sunyuze/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 459, in _load
    magic_number = pickle_module.load(f)
_pickle.UnpicklingError: unpickling stack underflow

Tags: 代码inpy模型importhomemodelline

热门问题