Python package to extract deep learning features
Project description
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]
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for imfeatures-0.0.2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7cb19b4944cc903b4aae80be9fabf15b205a8e7f64b525e76b0dbc4810d79c63 |
|
MD5 | 6c0ecee2abeaf6af1a105612adddb21a |
|
BLAKE2b-256 | a22a915ec56d86908a70eb02e3250eb73c611d910c3363fc5ffc4bd5e76f36e9 |