https://github.com/tusimple/resnet.mxnet
https://github.com/tusimple/resnet.mxnet
Last synced: 10 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/tusimple/resnet.mxnet
- Owner: TuSimple
- License: mit
- Created: 2017-08-22T03:27:04.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2018-09-03T06:46:01.000Z (over 7 years ago)
- Last Synced: 2025-04-06T03:02:59.704Z (10 months ago)
- Language: Python
- Size: 51.8 KB
- Stars: 8
- Watchers: 10
- Forks: 15
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# resnet.mxnet
This repository contains a [MXNet](https://github.com/apache/incubator-mxnet) implementation for the [ResNet-v2](https://arxiv.org/abs/1603.05027) and [ResNeXt](https://arxiv.org/abs/1611.05431) algorithms for image classification.
## Requirements and Dependencies
- Install [a modified MXNet](https://github.com/huangzehao/incubator-mxnet-bk) on a machine with CUDA GPU
- Install CUDNN v5 or v6
- Download the [ImageNet](http://image-net.org/download-images) dataset and creat pass through rec (following [tornadomeet's repository](https://github.com/tornadomeet/ResNet#imagenet) but using unchange mode)
## Train and Test
- Modify ```config/cfg.py``` by your setting
- ```python train.py```
- ```python test.py```
## Reference
[1] Kaiming He, et al. "Identity Mappings in Deep Residual Networks" arXiv:1603.05027 (2016).
[2] Xie, Saining, et al. "Aggregated residual transformations for deep neural networks." arXiv:1611.05431 (2016).
[3] Chen, Tianqi, et al. "Mxnet: A flexible and efficient machine learning library for heterogeneous distributed systems." arXiv:1512.01274 (2015).
[4] Torch training code and model provided by facebook, https://github.com/facebookresearch/ResNeXt
[5] MXNet training code provided by tornadomeet, https://github.com/tornadomeet/ResNet