写了Alexnet、resenetv1、mobilenetv2,基本的架构(backbone)使用slim的实现,slim是tensorflow的一个库,这个库实现了一些经典的网络架构,如Alexnet,resnet,inception系列等等。
在slim库中实现的Alexnet不同的是,没有使用LRN,并且把全连接层替换成卷积层。
- 代码实现多GPU训练
- 读入数据采用多进程的方式
- 损失函数采用交叉熵加L2正则
- 训练样本采用的是猫狗大战读入数据采用多进程的方式
数据集链接:https://pan.baidu.com/s/1GoFCwLs8hetPdPdqdCnRPA 密码:pxkx
Alexnet代码实现
网友评论