使用tensorflow mobilenet trainingin

时间:2018-01-21 11:02:26

标签: tensorflow

我使用 Tensorflow MobileNet retrain.py示例文件(在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py中)培训了分类网

当我使用经过训练的网络时,我设法一次只能运行一个图像(输入是张量流4d阵列形状[1,128,128,3])

不知道这是否意味着什么,但列车过程使用列车批次标志--train_batch_size=100

当我尝试对一批图像进行分类时(例如,tensorflow 4d数组形状[2,128,128,3]到'input'图层),我得到以下错误: ValueError: Cannot feed value of shape (2, 128, 128, 3) for Tensor 'input:0', which has shape '(1, 128, 128, 3)'

(在运行这个经过训练的网络会话之前,我正在使用预处理张量流会话来准备此网络的图像(调整大小,规范化等)

有谁知道我应该怎么做才能在这样的网络上运行一批图像,或者我如何配置retrain.py文件来创建允许批量运行的网络?

0 个答案:

没有答案