用大图像重新训练MobileNet

时间:2018-09-14 19:11:57

标签: python tensorflow machine-learning classification

我正在尝试使用我自己的图像(800x600px,彩色,所以我认为x3)来重新训练Tensorflow for Poets MobileNets教程,我认为问题是我使用的图像太大。

我已进入本教程的本页 https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#7

一切正常,但是一旦我放入新的图像路径和图像大小,就会出现以下错误: 错误:张量流:Mobilenet输入大小应为“ 224”,“ 192”,“ 160”或“ 128”,  但为架构“ mobilenet_0.50_480000”找到了“ 480000” 错误:tensorflow:无法识别架构标志

教程说

  

输入图像分辨率:128,160,192或224px。毫不奇怪,输入更高分辨率的图像会花费更多的处理时间,但会带来更好的分类精度。

     

模型的相对大小(占最大MobileNet的一部分):1.0、0.75、0.50或0.25。

您应该使用以下代码输入该信息:

    IMAGE_SIZE=224
    ARCHITECTURE="mobilenet_0.50_${IMAGE_SIZE}"

这是对移动网络的限制吗?它不能做更大的图像吗?另一种可能性是我的图像全是png,本教程全是jpeg,但是我看不到任何指定文件类型的地方。

任何建议将不胜感激

0 个答案:

没有答案