Implementation of vision models with their pretrained weights
Project description
tvmodels
The tvmodels library contains implementation of pretrained vision models in pytorch trained on ImageNet. Some of these models are not available in torchvision, so you can load them for here.
Installation
Run the following to install:
pip install tvmodels
Usage
from tvmodels.models import se_resnet50, resnet18
# Load the models
se_res_model = se_resnet50(pretrained=True)
res_model = resnet18(pretrained=True)
TODO
- SEResNet50
- SEResNet101
- SEResNet152
- SEResNeXt50_32_4
- SEResNeXt101_32_4
- SENet154
- ResNext50_32_4
- ResNext101_32_4
- ResNext101_32_8
- ResNext101_64_4
- ResNext152
- ResNet18
- ResNet34
- ResNet50
- ResNet101
- ResNet152
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tvmodels-0.0.2.tar.gz
(1.5 kB
view hashes)