Giter Club home page Giter Club logo

regnet's Introduction

Regnet

Introduction

Here is our pytorch pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"


Design space design

Comparison

P: Paper's. O: Our

Model [P/O] gflops [P/O] params [P/O] top-1 error
RerNetY-200MF 0.2/0.22 3.2/3.27 29.6/updating...
RerNetY-400MF 0.4/0.42 4.3/4.45 25.9/updating...
RerNetY-600MF 0.6/0.60 6.1/5.66 24.5/updating...
RerNetY-800MF 0.8/0.82 6.3/6.26 23.7/updating...

Best models


Top RegNetX models


Top RegNetY models

Datasets

We use Imagenet (ILSVRC2012) for all experiments, as stated in the paper.

Create a data folder under this repository,

cd {repo_root}
mkdir data
  • ImageNet: Download the ImageNet dataset and put the files as the following structure:
    data
    ├── train
    │   ├── n01440764
    │   └── n01443537
    │   └── ...
    │── val
    │   ├── n01440764
    │   └── n01443537
    │   └── ...
    
    Of course you could change this path to whatever you want based on your own preference, or mount it to a folder when using docker.

How to use our code

With our code, you can:

  • Train your model with default arguments by running python train.py -d path/to/image/root/folder
  • We also provide shell scripts which could be used to run training for first RegnetY models at ./scripts/. For example, if you want to train RegNetY 800MF, you could simply run ./scripts/RegnetY_800MF.sh

Requirements

  • python 3.7
  • pytorch 1.4
  • opencv (cv2)
  • pthflops
  • torchsummary

Updating (21/04/2020)

Complete all networks and training script. We are training RegnetY models and will update result soon.

References

regnet's People

Contributors

uvipen avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

regnet's Issues

Bug in building RegNetX

Hi,
Thanks for a GREAT repo!
I think there might be a bug in the creation of RegnetX here:

ls_group_width = ls_group_width.astype(np.int) * bottleneck_ratio

Why would you multiply the group_width by the bottleneck_ratio?

I will demonstrate through an example:

group_width = 16
block_width = 32
bottleneck_ratio = 2

With these set of parameters I would assume a bottleneck block will be created with 1/2 the channels in the bottleneck and 1 group convolution (i.e. standard convolution)
However: l.25 changes the groups to ls_group_width = ls_group_width.astype(np.int) * bottleneck_ratio => group_width = 32 making this block impossible and having the model FAIL!

Is this intentional or a bug?

Thank you very much

Performance for RegNetY

First, thanks for the awesome work of re-implementing RegNet.

I am having difficulty of reproducing the results for RegNetY-0.4GF. The configurations are taken from the original repo:

group_width = 8
initial_width = 48
slope = 27.89
quantized_param = 2.09
network_depth = 16

I only get 72.92 top-1 accuracy, but the original paper reported 74.2. Any thoughts on that?

Why my Acc@5 always shown 100?

I followed the README to create data folder.
data
├── train
│ ├── n0
│ └── n1
│ └── n2
│ └── n3
│ └── n4
│── val
│ ├── n0
│ └── n1
│ └── n2
│ └── n3
│ └── n4

I modified the NUM_CLASSES = 5 at src/config.py.
And all config are default (RegnetY 200MF).

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.