kpick.classifier.models package
Submodules
kpick.classifier.models.mobilenet module
- class kpick.classifier.models.mobilenet.InvertedResidual(*args: Any, **kwargs: Any)
Bases:
Module- forward(x)
- class kpick.classifier.models.mobilenet.MobileNetV2(*args: Any, **kwargs: Any)
Bases:
Module- forward(x)
- kpick.classifier.models.mobilenet.conv_1x1_bn(inp, oup)
- kpick.classifier.models.mobilenet.conv_bn(inp, oup, stride)
- kpick.classifier.models.mobilenet.make_divisible(x, divisible_by=8)
- kpick.classifier.models.mobilenet.mobilenet_v2(pretrained=True)
kpick.classifier.models.resnet module
- kpick.classifier.models.resnet.resnet(**kwargs)
Constructs a ResNet model.