Need help with ResNeXt-PyTorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

miraclewkf
152 Stars 63 Forks 1 Commits 5 Opened issues

Services available

!
?

Need anything else?

Contributors list

No Data

This is the PyTorch implement of ResNeXt (train on ImageNet dataset)

Paper: Aggregated Residual Transformations for Deep Neural Networks

Usage

Prepare data

This code takes ImageNet dataset as example. You can download ImageNet dataset and put them as follows. I only provide

ILSVRC2012_dev_kit_t12
due to the restriction of memory, in other words, you need download
ILSVRC2012_img_train
and
ILSVRC2012_img_val
.
├── train.py # train script
├── resnext.py # network of resnext
├── read_ImageNetData.py # ImageNet dataset read script
├── ImageData # train and validation data
    ├── ILSVRC2012_img_train
        ├── n01440764
        ├──    ...
        ├── n15075141
    ├── ILSVRC2012_img_val
    ├── ILSVRC2012_dev_kit_t12
        ├── data
            ├── ILSVRC2012_validation_ground_truth.txt
            ├── meta.mat # the map between train file name and label

Train

  • If you want to train from scratch, you can run as follows:
python train.py --batch-size 256 --gpus 0,1,2,3
  • If you want to train from one checkpoint, you can run as follows(for example train from
    epoch_4.pth.tar
    , the
    --start-epoch
    parameter is corresponding to the epoch of the checkpoint):
python train.py --batch-size 256 --gpus 0,1,2,3 --resume output/epoch_4.pth.tar --start-epoch 4

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.