我正在考虑风格转移。一个重要的部分是重用现有模型。目前我像这样导入 vgg 图:
net = vgg16.get_vgg_model()
name_layer = 'conv2_2/conv2_2:0'
feature_transformed_image = tf.import_graph_def(
net['graph_def'],
name='vgg',
input_map={'images:0': image},return_elements=[name_layer])
现在我想多次重复使用 vgg 模型(因为我需要获取多个图像的输出)。
不幸的是,多次运行这段代码会让我多次导入整个图(浪费 RAM 和时间)。
所以我的问题是:如何导入该图一次,然后通过多个输入重新使用它。
None
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)