使用CNN对Instagram上的图片进行分类
这是一个在Tensorflow上建立的CNN图片分类模型,训练和测试的图片均来自于对Instagram爬虫获得的图片.为了探究CNN深度以及batch normalization对于性能的影响,分别设计了deep model with batch normalization, simple model with batch normalizaion, simple model without batch normalization三个模型.
本项目的图片是通过在Instagram上爬虫得到的
由于文件大小的限制,本项目制作的tfrecords无法上传至github.可以
通过ins_image_input.py自己制作数据集,也可以从网盘下载本项目
的train.tfrecords和eval.tfrecords.
程序中的函数可以在Tensorflow官网
查找,将图像进行转化和读取可以参考convert_to_records.py
以及fully_connected_reader.py.
Tensorflow还有许多详细的例程可供参考