Skip to content

Latest commit

 

History

History
29 lines (25 loc) · 2.63 KB

File metadata and controls

29 lines (25 loc) · 2.63 KB

PyTorch models

Here you can find the pretrained models for the PyTorch framework. In the following table the information of the model, the model structure and the pretrained weights can be found.

To include a model in the API you have to download the model structure and place it in HOMEPATH/DeepClas4BioModels/PyTorch/Classification/model/modelName.py and download the weights and place it in HOMEPATH/DeepClas4BioModels/PyTorch/Classification/weights/modelName.pth.

Pretrained models

The list of the pretrained models are the following.

Model Model Structure Model Weights DataSet
AlexNet ImageNet
DenseNet121 ImageNet
DenseNet161 ImageNet
DenseNet169 ImageNet
DenseNet201 ImageNet
InceptionV3 ImageNet
ResNet18 ImageNet
ResNet34 ImageNet
ResNet50 ImageNet
ResNet101 ImageNet
ResNet152 ImageNet
ResNetKvasir Strucutre file Weights file ImageNet
SqueezeNet10 ImageNet
SqueezeNet11 ImageNet
VGG11 ImageNet
VGG13 ImageNet
VGG16 ImageNet
VGG19 ImageNet