1. 首页
  2. 编程语言
  3. Python
  4. 使用TensorFlow-Slim进行图像分类的实现

使用TensorFlow-Slim进行图像分类的实现

上传者: 2022-03-02 04:40:50上传 PDF文件 88.75 KB 热度 12次

参考 使用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趋势。官方提供了验证脚本。

用户评论