资源分类
》 资源 》算法》pre-resnet-gen-caffe

pre-resnet-gen-caffe

2020-02-21 | |  45 |   0 |   0

Pre-ResNet Generator

A simple python script to generate Caffe prototxt of pre-activation ResNet on the ImageNet (ILSVRC) dataset. See Identity Mappings in Deep Residual Networks and fb's torch implementation.

Usage

  1. Set the CAFFE_ROOT (line 8) properly.

  2. python2 net_def.py --depth 200

Depth can be 50, 101, 152, or 200. Will output resnet{depth}_trainval.txt respectively by default.

Note

  1. The script only serves as a network scaffold. More data augmentation schemes should be added to the data layers to achieve good performance.

  2. Some pre-activation BatchNorm layers could be further made inplace to save memory. Will double check the correctness of doing so.


上一篇：resnet.torch

下一篇：Identity-Mapping-ResNet-Lasagne

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • Klukshu-Sockeye-...

    KLUKSHU SOCKEYE PROJECTS 2016 This repositor...