Skip to content

tensorflow-slim下的inception_v3、inception_v4、inception_resnet_v2分类模型的数据制作、训练、评估、导出模型、测试

Notifications You must be signed in to change notification settings

7u5/aiface-tf1

 
 

Repository files navigation

tf-slim-inception

tensorflow-slim下的inception_v3、inception_v4、inception_resnet_v2分类模型的数据制作、训练、评估、导出模型、测试。 训练比较请参考:InceptionV3、InceptionV4图像分类训练与比较

相关截图

项目结构

{classname}是某个分类任务的名字,与 data/{classname} 样本目录相关 pos_{classname} 为该分类正例, neg_{classname}为复例

数据制作

image2tfrecord.py 数据制作请参考flowers

训练

  • train.py {classname}

评估

  • eval.py {classname}

可视化

  • show_train.py 训练
  • show_eval.py 评估

导出模型

  • export.py {classname}

测试

eval_single_img.py {classname} eval_single_dir.py {classname}

模型配置文件

  • train_inception.py 配置训练的参数(网络模型选择,训练次数,batch_size、指定GPU等)
  • config.py 配置文件

About

tensorflow-slim下的inception_v3、inception_v4、inception_resnet_v2分类模型的数据制作、训练、评估、导出模型、测试

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 77.5%
  • Jupyter Notebook 20.0%
  • Shell 2.5%