tensorflow搭建vgg16提取图像特征

github代码

https://github.com/ry/tensorflow-vgg16

https://github.com/machrisaa/tensorflow-vgg

预训练模型地址:

测试代码:

test_vgg16.py

import numpy as np
import tensorflow as tf

import vgg16
import utils

img1 = utils.load_image("./test_data/tiger.jpeg")
img2 = utils.load_image("./test_data/puzzle.jpeg")

batch1 = img1.reshape((1, 224, 224, 3))
batch2 = img2.reshape((1, 224, 224, 3))
fo=open('aa','w')
batch = np.concatenate((batch1, batch2), 0)
with tf.device('/gpu:1'):
    with tf.Session() as sess:
        images = tf.placeholder("float", [2, 224, 224, 3])
        feed_dict = {images: batch}

#        vgg = vgg16.Vgg16()
#        with tf.name_scope("content_vgg"):
#            vgg.build(images)

#        prob = sess.run(vgg.prob, feed_dict=feed_dict)
#        print(prob)
#        utils.print_pr
评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值