Ready-to-use implementations of some of the most common computer vision algorithms.
In PyTorch only!
Multi Task Cascaded Convolutional Neural Network (MTCNN) : A SOTA face and facial-landmark detection architecture. Check out this out for more details.
YOLOv3: The SOTA object detection algorithm. For more details, read the docs.
FaceNet: A Unified Embedding for Face Recognition and Clustering: One of the most popular architectures used for facial recognition. For more details, check here.
DEtection TRansformer (DETR): An end-to-end object detection architecture using transformers. For more details, check here.
Neural Style Transfer (NST): Transfer style from one Image into another. For more details, check here
Pyramid Scene Parsing Network (PSPNet): Instance Segmentation architecture that makes use of Pyramid Pooling Module for better results. For more details, check here.
For full list of architectures that has been ported or are in the process of being ported, check here.
$ git clone https://github.com/pranjaldatta/PyVision.git --single-branch --branch master
$ cd PyVision
$ pip install .
For contribution guidelines, please look here. Contributions are always welcome!
[ ] Populate with more architectures (obviously)
[x] Come up with an efficient way to make the repository minimal i.e. assets (like weights) will only be downloaded on as-you-need basis. All weights are hosted on SRM-MIC Google drive and downloaded using gdown
[x] Come up with an efficient way to ensure that heavy architecture specific dependecies are installed only when required. All heavy assets are installed only when model is being used.
Currently, its working only in pre-configured conda environment with all dependencies installed.
P.S. Star the repo if you liked the work!