AttributeError：模块“tensorflow.contrib.slim”没有属性“nets”

Question

我想使用tf-slim 中的内置resnet 进行快速实验。我是按照github中的README做的：

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))

但是遇到这样的错误：AttributeError: module 'tensorflow.contrib.slim' has no attribute 'nets'。我已经安装了最新版本的tensorflow-0.12.0。
我该如何解决这个问题？

Answer 1

像这样导入：

from tensorflow.contrib.slim.nets import resnet_v1

slim = tf.contrib.slim