Metadata-Version: 2.1
Name: tvmodels
Version: 0.0.5
Summary: Implementation of vision models with their pretrained weights
Home-page: https://github.com/rohitgr7/tvmodels
Author: Rohit Gupta
Author-email: rohitgr1998@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Development Status :: 5 - Production/Stable
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
Requires-Dist: torch (>=1.0.1)
Requires-Dist: numpy
Requires-Dist: requests

# tvmodels
The tvmodels library contains pretrained vision models in pytorch trained on ImageNet. Some of these models are available in [torchvision](https://pytorch.org/docs/stable/torchvision/index.html) but some are not, so you can load them for here.

## Installation
Run the following to install:
```python
pip install tvmodels
```

### Colab
If it shows `ModuleNotFoundError` on Google-colab use the following:
```python
!git clone https://github.com/rohitgr7/tvmodels.git
import sys
sys.path.append('/content/tvmodels')
```

## Usage
```python
from tvmodels.models import se_resnet50, resnet18

# Load the models
se_res_model = se_resnet50(pretrained=True)
res_model = resnet18(pretrained=True)
```

## Available models
- [x] ResNet(s)
- [x] ResNext(s)
- [x] SEResNet(s)
- [x] SEResNeXt(s)
- [x] SENet154
- [x] EfficientNets


