Modified from mmclassification.
Support 3D resnet pretraining(see configs/imagenet_pretrain/
).
Support LIDC-IDRI dataset(see configs/lidc/
).
Please refer to install.md for installation and dataset preparation.
Please see getting_started.md for the basic usage. There are also tutorials for finetuning models, adding new dataset, designing data pipeline, and adding new modules.
bash train.sh train
Results and models are available in the model zoo.
Supported backbones:
- ResNet
- ResNeXt
- SE-ResNet
- SE-ResNeXt
- RegNet
- ShuffleNetV1
- ShuffleNetV2
- MobileNetV2
- MobileNetV3