小编Bro*_*983的帖子

我已经使用 PyTorch 训练了一个模型并保存了一个状态字典文件。我已经使用下面的代码加载了预先训练的模型。我收到有关 RuntimeError: Error(s) in loading state_dict for VGG 的错误消息：

RuntimeError: Error(s) in loading state_dict for VGG:
    Missing key(s) in state_dict: "features.0.weight", "features.0.bias", "features.2.weight", "features.2.bias", "features.5.weight", "features.5.bias", "features.7.weight", "features.7.bias", "features.10.weight", "features.10.bias", "features.12.weight", "features.12.bias", "features.14.weight", "features.14.bias", "features.17.weight", "features.17.bias", "features.19.weight", "features.19.bias", "features.21.weight", "features.21.bias", "features.24.weight", "features.24.bias", "features.26.weight", "features.26.bias", "features.28.weight", "features.28.bias", "classifier.0.weight", "classifier.0.bias", "classifier.3.weight", "classifier.3.bias", "classifier.6.weight", "classifier.6.bias". 
    Unexpected key(s) in state_dict: "state_dict", "optimizer_state_dict", "globalStep", "train_paths", "test_paths". 

我正在遵循此网站上提供的说明：https://pytorch.org/tutorials/beginner/ saving_loading_models.html# saving-loading-model-across-devices

非常感谢

import argparse
import datetime
import glob
import os
import random
import shutil …

我正在运行下面的模型，在分割图像并尝试将图像分配回来后，我收到错误消息。

    sample[:, :, 0] = 0
TypeError: 'Image' object does not support item assignment

我尝试了不同的方法将图像分配回样本，但这没有帮助。任何帮助将不胜感激。非常感谢

这是我用来加载数据的代码：

import os

import numpy as np
import torch
from PIL import Image
from torch.utils.data import Dataset
from torchvision import transforms, utils
from torchvision.transforms import functional
#from torchvision.transforms import Grayscalei
import pandas as pd
import pdb
import cv2
from cv2 import imread
from cv2 import resize

class CellsDataset(Dataset):
    # a very simple dataset

    def __init__(self, root_dir, transform=None, return_filenames=False):
        self.root = root_dir
        self.transform = transform
        self.return_filenames …