wanggrun / Adaptively-Connected-Neural-Networks

A re-implementation of our CVPR 2019 paper "Adaptively Connected Neural Networks"
145 stars 29 forks source link

Update

Adaptively Connected Neural Networks

A re-implementation of our CVPR 2019 paper "Adaptively Connected Neural Networks" (https://arxiv.org/abs/1904.03579).

Guangrun Wang , Keze Wang, Liang Lin

Sun Yat-sen University (SYSU)

intro

Table of Contents

  1. Introduction
  2. ImageNet
  3. Cora
  4. Citation

Introduction

This repository contains the training & testing code of Adaptively Connected Neural Networks (ACNet) on ImageNet and Cora.

ImageNet

Model Top 5 Error Top 1 Error Download
ResNet50 6.9% 23.6% :arrow_down:
ResNet50-ACNet 6.4% 22.5% google drive: :arrow_down: and baidu pan: :arrow_down:
ResNet50-ACNet-pixel-aware 6.4% 22.5% google drive: :arrow_down: and baidu pan: :arrow_down:

python imagenet-resnet.py --gpu 0,1,2,3,4,5,6,7 --data [ROOT-OF-IMAGENET-DATASET] --log_dir [ROOT-OF-TRAINING-LOG-AND-MODEL]


+ Testing script:

cd cnn/dataset-aware/ or cd cnn/pixel-aware/

python imagenet-resnet.py --gpu 0,1,2,3,4,5,6,7 --data [ROOT-OF-IMAGENET-DATASET] --log_dir [ROOT-OF-TEST-LOG] --load [ROOT-TO-LOAD-MODEL] --eval


# Cora

Coming soon ...

# Citation

If you use these models in your research, please cite:

@inproceedings{wang2019adaptively,

      title={Adaptively Connected Neural Networks},

      author={Wang, Guangrun and Wang, Keze and Lin, Liang},

      booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},

      year={2019}}

# Dependencies
+ Python 2.7 or 3
+ TensorFlow >= 1.3.0
+ [Tensorpack](https://github.com/ppwwyyxx/tensorpack)
   The code depends on Yuxin Wu's Tensorpack. For convenience, we provide a stable version 'tensorpack-installed' in this repository. 

install tensorpack locally:

cd tensorpack-installed python setup.py install --user