/DeepMRD

Primary LanguagePython

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.

Getting Started

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.

Training

bash 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