训练好的模型参数可以使用.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的模型。