残差神经网络ResNet-tensorflow
已经在我的Github账号更新~
写的比较简单,传送门:
https://github.com/Sharpiless/RESNET-tensorflow
论文传送门:
https://arxiv.org/abs/1512.03385
原理详解在我的另一篇博客里:
残差神经网络ResNet系列网络结构详解:从ResNet到DenseNet
我用的是ImageNet的一个小数据集(原来是1000分类,小数据集使用了16个分类),使用别的数据集时注意修改一下read_data.py。