Agnostic Computer Vision Framework
Project description
An Agnostic Object Detection Framework
IceVision is the first agnostic computer vision framework to offer a curated collection with hundreds of high-quality pre-trained models from torchvision, MMLabs, and soon Pytorch Image Models. It orchestrates the end-to-end deep learning workflow allowing to train networks with easy-to-use robust high-performance libraries such as Pytorch-Lightning and Fastai
IceVision Unique Features:
-
Data curation/cleaning with auto-fix
-
Access to an exploratory data analysis dashboard
-
Pluggable transforms for better model generalization
-
Access to hundreds of neural net models
-
Access to multiple training loop libraries
-
Multi-task training to efficiently combine object detection, segmentation, and classification models
Installation
pip install icevision[all]
For more installation options, check our docs.
Important: We currently only support Linux/MacOS.
Quick Example: How to train the Fridge Objects Dataset
Happy Learning!
If you need any assistance, feel free to:
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 icevision-0.8.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e63654bdd7a97f616abb5b23edb6113f75587f9bfed2ffe6f640bd5979980f3c |
|
MD5 | 4c7b147682a32fb681bfc558f1f53e34 |
|
BLAKE2b-256 | 3195d7ea6ba72d6fba4bdd172916be4b8ce8ce3118e7c040ffa3d0d3494147a9 |