zjuzjz / cnn_models_in_tensorflow

使用TensorFlow自己搭建一些经典的CNN模型,并使用统一的数据来测试效果。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cnn_models_in_tensorflow

使用TensorFlow自己搭建一些经典的CNN模型,并使用统一的数据来测试效果。

模型:

  • AlexNet, 2012 (paper)
  • GoogLeNet, 2014.09 (paper)
  • Vgg, 2014.09 (paper)
  • Inception_v2, 2015.02 (paper)
  • Inception_v3, 2015.12 (paper)
  • ResNet, 2015.12 (paper)

数据:

数据原下载地址在这里,tgz文件大小约218M,包含5类图片大小不确定的3通道花朵图片,各类数目不等,共计3670张。但由于本人笔记本限制,实验时只取了部分图片,即根目录下datasets中文件,5类各取400张,共2000张。实验时取测试集比例为25%,各100张。

文件说明:

  • datasets:数据集文件夹;
  • graphs:模型在tensorboard中的graph,部分graph由于尺寸限制只截取了代表部分;
  • images:一些供ipynb文件调用的可视化或者其他辅助图片文件夹;
  • models:搭建模型文件夹;
  • *.ipynb:实验记录文档,一些文字记录、图片说明,包括迭代记录、模型说明和对比等;
  • alexnet.py:最开始的第一个模型,此处意义不大,在models文件夹中也有;

About

使用TensorFlow自己搭建一些经典的CNN模型,并使用统一的数据来测试效果。


Languages

Language:Jupyter Notebook 91.9%Language:Python 8.1%