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.9.0a1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7da6d10400b5eee27415a4687fa768f2976c536ef20aeebcda88ac65f1f7124f |
|
MD5 | 94f069a6d7d18498db8760358afca4e4 |
|
BLAKE2b-256 | bd84070d6b6fa9bd6116078247d82e4086c45b7947e3e8a24422b914dcdf7f07 |