基于TF的迁移学习VGG16网络,并进行图片的测试。
vgg16.py文件1.一开始在通过vgg =Vgg16()的调用,会自动初始化,在初始化过程中,会找到目录下的vgg16.npy文件,也就是在imagenet上训练好的权重文件,然后通过np.load()完成权重文件的加载。代码如下:class Vgg16: def __init__(self, vgg16_npy_path=None): if vgg16_npy_pat...
原创
2018-05-04 16:02:44 ·
2552 阅读 ·
1 评论