使用训练好的模型参数

训练好的模型参数可以使用.npy形式储存与读取,具体细节参考VGG-16的程序
命名好卷积层,每一个卷积层list中含有两个成分(权重和偏置)
读取:data_dict = loadWeightsData('./tensorflow_vgg/vgg16.npy')

def loadWeightsData(vgg16_npy_path=None):
    if vgg16_npy_path is None:
        path = inspect.getfile(Vgg16)
        path = os.path.abspath(os.path.join(path, os.pardir))
        path = os.path.join(path, "vgg16.npy")
        vgg16_npy_path = path
        print (vgg16_npy_path)
    return np.load(vgg16_npy_path, encoding='latin1').item()

data_dict = np.load(vgg16_npy_path, encoding=‘latin1’).item()
在读取训练好的参数时仍然需要再次搭建VGG-16的模型。

猜你喜欢

转载自blog.csdn.net/weixin_41855385/article/details/84475540