VGG_MEAN = [103.939, 116.779, 123.68] class Vgg19: def __init__(self, vgg19_npy_path=None): if vgg19_npy_path is None: path = inspect.getfile(Vgg19) path = os.path.abspath(os.path.join(path, os.pardir)) path = os.path.join(path, "vgg19.npy") vgg19_npy_path = path print(vgg19_npy_path) self.data_dict = np.load(vgg19_npy_path, encoding='latin1').item() print("npy file loaded")
时间: 2024-04-03 20:37:04 浏览: 162
‘vgg_generated_120.i’and ‘boostdesc_bgm.i’No such file or directory.rar
这段代码进一步完善了之前定义的Vgg19类,其中np.load()函数用于载入预先训练好的VGG19模型的参数,encoding='latin1'参数用于指定读取文件时使用的编码方式。该函数返回一个字典类型的对象,包含了VGG19模型中所有层的权重和偏置等参数。这些参数被存储在self.data_dict变量中,并在载入完成后打印了一条提示信息。
阅读全文