2016-12-04 11 views
2

tf-slimに内蔵されたresnetを使いたいです。AttributeError:モジュール 'tensorflow.contrib.slim'に 'nets'属性がありません

import tensorflow as tf 
import tensorflow.contrib.slim as slim 

resnet = tf.contrib.slim.nets.resnet_v1 

mnist = input_data.read_data_sets('MNIST_data', one_hot=True) 
x = tf.placeholder("float", shape=[None, 784]) 
y_ = tf.placeholder("float", shape=[None, 10]) 
pred = resnet.resnet_v1_50(x) 

cross_entropy = -tf.reduce_sum(y_ * tf.log(pred)) 

しかし、このようなエラーました:私はgithubREADMEに応じなかったAttributeError: module 'tensorflow.contrib.slim' has no attribute 'nets'を。私はすでにtensorflow-0.12.0の最新バージョンをインストールしています。
この問題を解決するにはどうすればよいですか?このような

答えて

0

インポート:

from tensorflow.contrib.slim.nets import resnet_v1 

slim = tf.contrib.slim 
関連する問題