Alexnet、resenetv1、mobilenetv2网络实现,基本的架构(backbone)使用slim的实现,slim是tensorflow的一个库,这个库实现了一些经典的网络架构,如Alexnet,resnet,inception系列等等。
在slim库中实现的Alexnet不同的是,没有使用LRN,并且把全连接层替换成卷积层。
- 代码实现多GPU训练
- 读入数据采用多进程的方式
- 损失函数采用交叉熵加L2正则
- 训练样本采用的是猫狗大战读入数据采用多进程的方式
数据集链接:https://pan.baidu.com/s/1GoFCwLs8hetPdPdqdCnRPA 密码:pxkx
Alexnet代码实现
参考:
https://www.jianshu.com/p/4c261534ab3f
https://github.com/liuheng92/Alexnet-tensorflow
https://blog.youkuaiyun.com/matrix_space/article/details/88776860
https://www.jianshu.com/p/1cf3b543afff?utm_source=oschina-app
https://blog.youkuaiyun.com/zj360202/article/details/78623567
https://blog.youkuaiyun.com/weixin_41803874/article/details/81567791