Image classification library for easily and quickly deploying models and training classifiers
Project description
TensorImage
TensorImage is an open source library designed to make training and deploying image classification models easy.
Features
-
Cluster training: automatically compare the performance of multiple trainers, speeding up the process of hyperparameter tuning and feature engineering, as there is no need to do it manually
-
Multithreaded training: by default, all training operations are run in 10 threads to make training models faster
-
Built-in image data augmentation operations, which can be used for feature engineering:
- Image flipping
- Salt-pepper noise
- Random brightness
- Random contrast
- Random hue
- Random saturation
- Gaussian blur
- Colour filtering
-
Workspace organization: all datasets, trained models, and internal metadata files are stored automatically inside a workspace directory, where you can quickly find any files you need
-
Large-scale image classification: deploy trained models on thousands of images, with predictions for all images being stored in your workspace directory
Upcoming features
-
More data augmentation operations to apply on images:
- Affine/perspective transformations
- Random zooming
- Random cropping
- Individual pepper and salt noise
- More image blurring techniques:
- Median blur
- Average blur
- Motion blur
- Bilateral blur
- Translation
-
Option to apply different data augmentation operations at once, e.g: instead of only applying gaussian blur, to be able to apply gaussian blur, pepper salt noise and random contrast at once, not uniquely separately
-
Model inference for individual/batches of images for real-time prediction without writing on disk
-
Real-time training from individual/batches of images without reading from disk, automatically training the model from new data, linked to real-time inference without having to store the model in disk (with option to store available)
Installation
From the terminal:
$ pip3 install tensorimage
Documentation
You can view TensorImage's documentation here.
Support
If you are experiencing any errors or bugs, please report them in the issues section or contact us at tensor.image2@gmail.com
Contributing
If you have any ideas for features that should be added to TensorImage, please feel free to fork TensorImage and open a pull request.
License
TensorImage is licensed under the MIT license.
Project details
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 TensorImage-2.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | db7dabe3b97023799440618dfb38ef3f6bbbc480468df31844ec3582eac27159 |
|
MD5 | 94f1085e67aba6ef18944752c62a3df5 |
|
BLAKE2b-256 | 54126eda020f58c91aac728b9095c7c711ffb9b159b699a91c57ac065f9dd217 |