参考 使用TensorFlow-Slim进行图像分类安装TensorFlow参考 如在Ubuntu下安装TensorFlow with GPU support, python 2.7版本下载TF-slim图像模型库准备数据有不少公开数据集，这里以官网提供的Flowers为例。读入数据官方提供了读入Flowers数据集的代码models/slim/datasets/flowers.py，同样这里也是参考并修改成能读入上面定义的通用数据集。如需要自定义模型，则参考官方提供的模型并放在对应的文件夹即可。fine-tune所有层把--checkpoint_exclude_scopes和--trainable_scopes删掉。其它参数可删掉用默认值或自行修改。使用自己的数据则需要修改models/slim/train_image_classifier.py：把修改为把修改为在后加入训练时执行以下命令即可：可视化log可一边训练一边可视化训练的log，可看到Loss趋势。官方提供了验证脚本。
暂无评论