tvmodels
The tvmodels library contains pretrained vision models in pytorch trained on ImageNet. Some of these models are available in torchvision but some are not, so you can load them for here.
Installation
Run the following to install:
pip install tvmodels
Colab
If it shows ModuleNotFoundError on Google-colab use the following:
!git clone https://github.com/rohitgr7/tvmodels.git
import sys
sys.path.append('/content/tvmodels')
Usage
from tvmodels.models import se_resnet50, resnet18
# Load the models
se_res_model = se_resnet50(pretrained=True)
res_model = resnet18(pretrained=True)
Available models
- ResNet(s)
- ResNext(s)
- SEResNet(s)
- SEResNeXt(s)
- SENet154
- EfficientNets
1 maintainer
Metadata
1 maintainer
Release files for tvmodels 0.0.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| tvmodels-0.0.7.tar.gz | 7.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tvmodels-0.0.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 18.6 kB
Release files / tvmodels-0.0.7.tar.gz
| Download URL | tvmodels-0.0.7.tar.gz |
|---|---|
| Size | 7.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
c9466812e5b1a328eb937da3da47ef8d98f74253c1371eb020cfa9db438721a6
|
|
BLAKE2b-256 checksum How to use checksums |
821cf86db61ae7d3003e14db8d2fff5ab4de7d7216678f9c5caea013ac97e082
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.2
|
Release files / tvmodels-0.0.7-py3-none-any.whl
| Download URL | tvmodels-0.0.7-py3-none-any.whl |
|---|---|
| Size | 11.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
b46f5429bdf2f3771f9a6151282a2cc6f0414f0ed974de6e0dc00798d708e67b
|
|
BLAKE2b-256 checksum How to use checksums |
98cd44798336d94dc8c8faacdda004e734e5dbcc3c577449a0c756cdcd994aad
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.2
|
