Python tf slim inceptionv3输出错误
我想用tf slim的网络预测图像。 但我得到了inceptionv3的随机结果。 对于resnet50,一切正常 resnet50:Python tf slim inceptionv3输出错误,python,tensorflow,deep-learning,resnet,tensorflow-slim,Python,Tensorflow,Deep Learning,Resnet,Tensorflow Slim,我想用tf slim的网络预测图像。 但我得到了inceptionv3的随机结果。 对于resnet50,一切正常 resnet50: import tensorflow as tf import cv2 import numpy as np import tensorflow.contrib.slim.nets as nets slim = tf.contrib.slim with tf.device('/gpu:1'): inputs = tf.placeholder(tf.flo
import tensorflow as tf
import cv2
import numpy as np
import tensorflow.contrib.slim.nets as nets
slim = tf.contrib.slim
with tf.device('/gpu:1'):
inputs = tf.placeholder(tf.float32, shape=[None,299,299,3])
with slim.arg_scope(nets.resnet_v1.resnet_arg_scope()):
features,net = nets.resnet_v1.resnet_v1_50(inputs=inputs, num_classes=1000)
saver = tf.train.Saver()
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.allow_soft_placement=True
with tf.Session(config=config) as sess:
saver.restore(sess, 'weights/resnet_v1_50.ckpt')
img = cv2.imread('images/dog_ball.jpg')
img = cv2.resize(img,(299,299))
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = img/255.0
curr_features, curr_net = sess.run([features, net], feed_dict={inputs: [img,img, img]})
for curr_feature in curr_features:
f_ind = np.argsort(curr_feature[0][0])[-4:] # resnet50v1
for i in f_ind:
print i
print ' '
但是如果我尝试《盗梦空间3》,它就不起作用了。
即使图像相同,结果也不相同。
首先我想,重量没有正确加载,但一切看起来都很好
接收v3:
import tensorflow as tf
import cv2
import numpy as np
import tensorflow.contrib.slim.nets as nets
slim = tf.contrib.slim
with tf.device('/gpu:1'):
inputs = tf.placeholder(tf.float32, shape=[None,299,299,3])
with slim.arg_scope(nets.inception.inception_v3_arg_scope()):
features,net = nets.inception.inception_v3(inputs=inputs, num_classes=1001)
saver = tf.train.Saver()
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.allow_soft_placement=True
with tf.Session(config=config) as sess:
saver.restore(sess, 'weights/inception_v3.ckpt')
img = cv2.imread('images/dog_ball.jpg')
img = cv2.resize(img,(299,299))
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = img/255.0
curr_features, curr_net = sess.run([features, net], feed_dict={inputs: [img,img, img]})
for curr_feature in curr_features:
f_ind = np.argsort(curr_feature)[-4:] # inceptionv3
for i in f_ind:
print i
print ' '
你知道,我的错误在哪里吗?找到了答案
如果您有相同的问题,请写:
features,net = nets.inception.inception_v3(inputs=inputs, num_classes=1001, is_training=False)