mednet.config.classify.modelsΒΆ

Modules

alexnet

AlexNet network architecture, to be trained from scratch.

alexnet_pretrained

AlexNet network architecture, pre-trained with ImageNet weights.

cnn3d

Simple CNN for 3D image classification, to be trained from scratch.

densenet

DenseNet, to be trained from scratch.

densenet_pretrained

DenseNet, to be fine-tuned.

logistic_regression

Logistic regression classifier, to be trained from scratch.

mlp

Feedforward shallow MLP for binary classification.

pasa

Simple CNN for Tuberculosis Detection, to be trained from scratch.