056cb9970e92

向楼主请教：想在自己的CPU笔记本上调试程序，然后到服务器上去跑。在CPU上跑完后拿到TensorFlow-GPU上去跑需要更改哪里呢？

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

博主您好,我想把模型换成resnet50,残差网络的代码是您的另外一篇文章,应该对哪里进行对残差网络程序做哪些修改呢?

TensorFlow 训练多任务多标签模型

在学习了使用 TensorFlow 的 CNN 进行图像分类之后，现在对这些方法做一个简单的拓展，即来处理多任务多标签的情形。为了便于说明，我们假设现在要对 0-9 这 10...

公输睚信

22414 36 23 1

056cb9970e92

您好 我刚刚试了一下换成mobilenet但是一直有问题，请问你做了哪些修改

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 谢谢你提供的帮助和鼓励

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 解决了 感谢博主 期待你进一步更新博客。但是预训练参数的导入一直还是有问题，

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 怎么能解决这个问题呢？我已经把训练程序里面的类别数目更改了 但是还是报错

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 博主你好 这是具体的错误信息
InvalidArgumentError (see above for traceback): Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Assign requires shapes of both tensors to match. lhs shape= [2048,5] rhs shape= [2048,61]
[[Node: save/Assign_2 = Assign[T=DT_FLOAT, _class=["loc:@Predict/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Predict/weights, save/RestoreV2/_5)]]

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 现在不在实验室 明天具体给您发一下错误信息 大概是说[2046.61]和[2046.5]不对应吧

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

@公输睚信 能具体说一下嘛？怎么才能禁止第一层参数导入

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

博主你好，我的预训练模型参数一直没有导入进去，是不是我因为我的训练图片是灰度图，如果是的话我怎样处理才能把预训练模型导入进去

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92

flags.DEFINE_integer('num_samples', 3273, 'Number of samples.')
flags.DEFINE_integer('num_classes', 61, 'Number of classes')
楼主您好，在您github给的程序中这个num-classes设置为61 而当我把他改成我的类别5时就会报错

TensorFlow 使用预训练模型 ResNet-50

升级版见：TensorFlow 使用 tf.estimator 训练模型（预训练 ResNet-50）。 前面的文章已经说明了怎么使用 TensorFlow 来构建、训练、保...

公输睚信

102493 235 53 5

056cb9970e92 ·

作者您好，读了您这篇文章收获很大，但是我遇到一点麻烦，还请您指点，文章使用slim读取tfrecord的多有操作都是针对jpg、png等格式的二维图像，在使用slim读取tfrecord文件时必须指定文件的format，如果原始数据是一个三维数据例如MRI图像，这种图像是三维矩阵，并且在slim中没有对应的的格式，这种情况该如何生成和读取tfrecord文件？

4cfb6691a22b 评论自TensorFlow-slim 训练 CNN 分类模型（续）