机器学习

tensorflow实现Alexnet resnetv1 mob

2018-09-26  本文已影响0人  ce0b74704937

写了Alexnet、resenetv1、mobilenetv2,基本的架构(backbone)使用slim的实现,slim是tensorflow的一个库,这个库实现了一些经典的网络架构,如Alexnet,resnet,inception系列等等。

在slim库中实现的Alexnet不同的是,没有使用LRN,并且把全连接层替换成卷积层。

  1. 代码实现多GPU训练
  2. 读入数据采用多进程的方式
  3. 损失函数采用交叉熵加L2正则
  4. 训练样本采用的是猫狗大战读入数据采用多进程的方式

数据集链接:https://pan.baidu.com/s/1GoFCwLs8hetPdPdqdCnRPA 密码:pxkx
Alexnet代码实现

上一篇下一篇

猜你喜欢

热点阅读