ResNet网络搭建及使用

放假前需要测试ResNet的准确度,本想着这么经典而又朴素的网络,因为有很多现成的代码可以用于学习,但大多数都只是对于网络的关键架构进行介绍以及代码实现,而对于新手来说需要一个完整可跑的框架,寻找着实不易。

在git上寻找了很久,终于找到了这个可用的框架,今天对于其中的重要文件进行记录,也作为个人的回忆;

git地址:https://github.com/wenxinxu/resnet-in-tensorflow

该代码原本使用的是cifar数据集,并且需要从网络上进行下载,图片大小格式默认为32*32,对于我们主流的100*100以及224*224的图像并不支持,此外,该代码默认使用TfRecord数据格式进行解析。因此我将代码进行了改进以及调整,使得任意数据集都可以通过这个ResNet网络进行测试,稍后会将修改过的代码版本链接发出来。

 

你可能感兴趣的:(python,tensorflow)