项目作者: CuthbertCai

项目描述 :
使用CNN对Instagram上的图片进行分类
高级语言: Python
项目地址: git://github.com/CuthbertCai/INS_Classify.git
创建时间: 2017-07-12T10:57:29Z
项目社区:https://github.com/CuthbertCai/INS_Classify

开源协议:

下载


INS_CLassify

这是一个在Tensorflow上建立的CNN图片分类模型,训练和测试的图片均来自于对Instagram爬虫获得的图片.为了探究CNN深度以及batch normalization对于性能的影响,分别设计了deep model with batch normalization, simple model with batch normalizaion, simple model without batch normalization三个模型.

  • ins_image_input.py用于将JPG格式的图片转化为tfrecords文件,
    为训练和测试提供图片.
  • ins_train.py用于对模型进行训练,并记录checkpoint
  • ins_eval.py用于对模型进行测试,记录测试时的准确率
  • ins_model.py,ins_small_model.py,ins_small_model_bn
    分别设计了三种CNN模型用于训练和测试

图片来源

本项目的图片是通过在Instagram上爬虫得到的

由于文件大小的限制,本项目制作的tfrecords无法上传至github.可以
通过ins_image_input.py自己制作数据集,也可以从网盘下载本项目
train.tfrecordseval.tfrecords.

参考资料

程序中的函数可以在Tensorflow官网
查找,将图像进行转化和读取可以参考convert_to_records.py
以及fully_connected_reader.py.
Tensorflow还有许多详细的例程可供参考