Introduction

Modified from mmclassification. Support 3D resnet pretraining(see configs/imagenet_pretrain/). Support LIDC-IDRI dataset(see configs/lidc/).

Installation

Please refer to install.md for installation and dataset preparation.

Training

bash scripts/train.sh train

Benchmark and model zoo

Results and models are available in the model zoo.

Supported backbones:

• ResNet
• ResNeXt
• SE-ResNet
• SE-ResNeXt
• RegNet
• ShuffleNetV1
• ShuffleNetV2
• MobileNetV2
• MobileNetV3