Model
Supported classification model architectures.
Usage
Model(
_,
size=224,
)Attributes
| Name | Description |
|---|---|
| alexnet | |
| alexnet_att | |
| densenet121 | |
| densenet161 | |
| densenet169 | |
| densenet201 | |
| efficientnetb3 | |
| efficientnetb3s | |
| efficientnetb4 | |
| efficientnetb4s | |
| googlenet | |
| googlenet_att | |
| resenext50_32x4d | |
| shufflenet2x1_0 | |
| shufflenet2x1_5 | |
| shufflenet2x2_0 | |
| simplegooglenet | |
| simplegooglenet_att | |
| simplegooglenet_ch_att | |
| simplegooglenet_ch_sp_att | |
| size | Return the expected square input size in pixels. |
| squeezenet1_0 |
alexnet
alexnet="alexnet"
alexnet_att
alexnet_att="alexnet_att"
densenet121
densenet121="densenet121"
densenet161
densenet161="densenet161"
densenet169
densenet169="densenet169"
densenet201
densenet201="densenet201"
efficientnetb3
efficientnetb3=("efficientnetb3", 300)
efficientnetb3s
efficientnetb3s=("efficientnetb3s", 300)
efficientnetb4
efficientnetb4=("efficientnetb4", 380)
efficientnetb4s
efficientnetb4s=("efficientnetb4s", 380)
googlenet
googlenet="googlenet"
googlenet_att
googlenet_att="googlenet_att"
resenext50_32x4d
resenext50_32x4d="resenext50_32x4d"
shufflenet2x1_0
shufflenet2x1_0="shufflenet2x1_0"
shufflenet2x1_5
shufflenet2x1_5="shufflenet2x1_5"
shufflenet2x2_0
shufflenet2x2_0="shufflenet2x2_0"
simplegooglenet
simplegooglenet="simplegooglenet"
simplegooglenet_att
simplegooglenet_att="simplegooglenet_att"
simplegooglenet_ch_att
simplegooglenet_ch_att="simplegooglenet_ch_att"
simplegooglenet_ch_sp_att
simplegooglenet_ch_sp_att="simplegooglenet_ch_sp_att"
size
Return the expected square input size in pixels.
size: int
squeezenet1_0
squeezenet1_0="squeezenet1_0"
Methods
| Name | Description |
|---|---|
| __init__() | Store the expected input image size. |
| __new__() | |
| __str__() | Return the command-line value. |
__init__()
Store the expected input image size.
Usage
__init__(_, size=224)__new__()
Usage
__new__(value)__str__()
Return the command-line value.
Usage
__str__()