Autolabel is an image labeling tool using Neural Network
Project description
Autolabel
Autolabel is an image labeling tool. Currently images are labeled using ResNet18-152 implemented by pytorch. Autolabel can be used as a cli tool or as a library.
Installation
python setup.py install
Command line usage
See --help
for a command overview
Usage: autolabel [OPTIONS] [IMAGES]...
Options:
--batch-size INTEGER
--sep TEXT Separator
--top INTEGER
-o, --output FILENAME Output file
-m, --model [resnet18|resnet34|resnet50|resnet101|resnet152]
--help Show this message and exit.
In the simplest form this mean:
autolabel image.jpg
Autolabel supports reading file names from STDIN:
find /myimages -type f -iname '*.jpg' | autolabel
Library usage
from autolabel.image import ImageListDataset
from autolabel.classifier.resnet import Resnet18Classifier
from pathlib import Path
classifier = Resnet18Classifier()
images = [Path('/path/to/image.jpg'), Path('/path/to/another/image.png')]
dataset = ImageListDataset(images)
res = classifier.predict(dataset, top=top)
for p, decoded in res.items():
print(p, decoded)
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
autolabel-0.0.1.tar.gz
(21.8 kB
view hashes)
Built Distributions
autolabel-0.0.1-py3.8.egg
(83.7 kB
view hashes)
autolabel-0.0.1-py3-none-any.whl
(36.9 kB
view hashes)
Close
Hashes for autolabel-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fd7306779ba0fd24aba560f5a0425efb0b11c959e271a366a15ee128c34f4c17 |
|
MD5 | 288788207506c2296b253ef56fa866f2 |
|
BLAKE2b-256 | 0d6da0fa34ad9ae24307ddfc54b99ff34ddae47dac365ce5cb6f04ef5fa09642 |