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.7.1a1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4b8526d5c7663fc660047312c62af61553d40385f79f35c8eeab4e87edee63c1 |
|
MD5 | 774bab978be2e985dce5f23dfce864bf |
|
BLAKE2b-256 | ece96b8a4964e0c8848ac0e2e30ca9dd0b23bd078b7f5b7311c1b10925782e9a |