semvec/pypi-packages huggingface dataset:
imfeatures 0.0.2
Python package to extract deep learning features
License: BSD

imfeatures
===============================

Overview
--------

Minimalistic python package to extract deep learning features from a wide variety of pretrained models in Pytorch.


Installation / Usage
--------------------

Dependencies:

- pytorch (v >= 1.0.0)
- torchvision


To install use pip:

    $ pip install imfeatures


Or clone the repo:

    $ git clone https://github.com/resbyte/imfeatures.git

    $ python setup.py install


Example
-------

- Imports 

```python
import imfeatures
import torch
```

- create feature extractor, here `resnet50`, with pretrained weights 

```python
feature_extractor = imfeatures.Features('resnet50',pretrained=True)
```

- random image of size `224x224x3` 

```python
x = torch.randn([1,3,224,224])
```

- features

```python 
features = feature_extractor(x)
print(features.shape)
```

Output features will be of shape : `[1, 2048, 1, 1]`

Author
-------

Abhinav Dadhich