Top 23 Python Computer Vision Projects

• Face Recognition

  15 41,739 2.1 Python

  The world's simplest facial recognition api for Python and the command line

  Project mention: I need to create a python program which uses camera to take photos and feed the identity data into a database. I have a Raspberry pi 3a+ 512mb ram. Is it possible and any tips? | reddit.com/r/RASPBERRY_PI_PROJECTS | 2021-09-26

  sudo pip3 install face_recognition git clone --single-branch https://github.com/ageitgey/face_recognition.git

• pytorch-CycleGAN-and-pix2pix

  3 15,969 5.9 Python

  Image-to-Image Translation in PyTorch

  Project mention: I made a 3d topographic map based on my recent civ6 game | reddit.com/r/civ | 2021-08-03

  pix2pix algorithm is used for translating Civ6Maps to heightmaps. Synthesized terrain was rendered in blender.

• Activeloop.ai

  activeloop.ai

  sponsored

  Optimize your datasets for ML. Goodbye, boilerplate code - the fastest dataset optimization and management tool for computer vision.

  

• EasyOCR

  12 12,810 8.0 Python

  Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

  

  Project mention: How would one go about recognizing a text (date) in an image? | reddit.com/r/MLQuestions | 2021-09-07

  Try running https://github.com/JaidedAI/EasyOCR, which reads all the text from a given image. Then loop over the words read and see if it reads your date properly or if it splits it into multiple words. Even if it does this, you can easily find the year (you might need to do some postprocessing on what the ocr reads, i.e transform l to 1). The nice thing is that you get the bounding boxes of where the tect was read so you can use some sort of postprocessing on bbox locations and words read to find the month amd day (i.e. I read 25 at the same y index as the yesr but the x index is to the left, so its probably the day). Good luck :)

• facenet

  5 12,058 0.0 Python

  Face recognition using Tensorflow

  Project mention: CompreFace - Free and open-source self-hosted face recognition system from Exadel | reddit.com/r/selfhosted | 2021-05-07

  As for me, openface is already outdated - the latest release was in 2016. If you look for a library, the easiest to use is ageitgey/face_recognition. The more accurate libraries are davidsandberg/facenet and deepinsight/insightface.

• jina

  94 11,559 10.0 Python

  Cloud-native neural search framework for 𝙖𝙣𝙮 kind of data

  

  Project mention: DAE think it's really cool that we have open-source software tools to build our own search engines? | reddit.com/r/DoesAnybodyElse | 2021-10-11

  if you'd like to skip the tutorial and dive head-on: https://github.com/jina-ai/jina

• d2l-en

  1 11,202 9.8 Python

  Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 200 universities.

  

  Project mention: I created a way to learn machine learning through Jupyter | reddit.com/r/learnmachinelearning | 2021-04-30

  There are actually some online books and courses built on Jupyter Notebook ([Dive to Deep Learning Book](https://github.com/d2l-ai/d2l-en) for example). However yours is more detail and could really helps beginners.

• datasets

  5 10,108 9.9 Python

  🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

  

  Project mention: Datasets: A Community Library for Natural Language Processing | news.ycombinator.com | 2021-09-08

• Nanos

  github.com/nanovms

  sponsored

  Run Linux Software Faster and Safer than Linux with Unikernels.

  

• vision

  4 10,014 9.8 Python

  Datasets, Transforms and Models specific to Computer Vision

  

  Project mention: Faster RCNN Training help | reddit.com/r/computervision | 2021-09-14

  I’m afraid that I don’t have time to look at your code right now, but you might find the torchvision implementation useful.

• fashion-mnist

  7 9,507 4.2 Python

  A MNIST-like fashion product database. Benchmark :point_down:

  

  Project mention: A New Google AI Research Study Discovers Anomalous Data Using Self Supervised Learning | reddit.com/r/ArtificialInteligence | 2021-09-07

  New Google AI research introduces a 2-stage framework that uses recent progress on self-supervised representation learning and classic one-class algorithms. This framework is simple to train and shows SOTA performance on various benchmarks, including CIFAR, f-MNIST, Cat vs. Dog, and CelebA. Following that, they offer a novel representation learning approach for a practical industrial defect detection problem using the same architecture. On the MVTec benchmark, the framework achieves a new state-of-the-art.

• labelme

  3 7,343 7.0 Python

  Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).

  Project mention: Convert Yolov3 annotation to labelme | reddit.com/r/deeplearning | 2021-04-25

  Ref. - https://github.com/wkentaro/labelme/

• meshroom

  56 6,997 9.5 Python

  3D Reconstruction Software

  

  Project mention: How to update cameraSensors.db with Sony ZV-1's data ? | reddit.com/r/Meshroom | 2021-10-06

• label-studio

  21 6,709 9.8 Python

  Label Studio is a multi-type data labeling and annotation tool with standardized output format

  

  Project mention: Offline text-labeling tool | reddit.com/r/LanguageTechnology | 2021-10-16

  labelstudio works offline. You can start a server locally.

• Serpent.AI

  3 6,052 0.0 Python

  Game Agent Framework. Helping you create AIs / Bots that learn to play any game you own!

  

  Project mention: I'm on a 64 bit win10 pc and want to make a tas for a unity game, that is what I have. How do I make a tas | reddit.com/r/TAS | 2021-05-27

  i cant. is there any way https://github.com/SerpentAI/SerpentAI would work. the game is entirely mouse movements.

• vit-pytorch

  4 5,999 8.5 Python

  Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

  Project mention: [D] Training vision transformers on a specific dataset from scratch | reddit.com/r/MachineLearning | 2021-08-06

  lucid rains VI has all of what you may need in a clean API

• pix2pixHD

  3 5,476 0.0 Python

  Synthesizing and manipulating 2048x1024 images with conditional GANs

  

  Project mention: [P] I made FaceShop! Instance segmentation + CGAN for editing faces (badly) | reddit.com/r/MachineLearning | 2021-09-25

  Pix2PixHD (from DeepSIM)

• Kornia

  5 5,063 9.3 Python

  Open Source Differentiable Computer Vision Library

  

  Project mention: Preprocessing for NN on GPU | reddit.com/r/computervision | 2021-10-04

• BackgroundMattingV2

  1 4,858 6.0 Python

  Real-Time High-Resolution Background Matting

  Project mention: [D] AI Background Removal: a quick comparison between RVM & BGMv2 | reddit.com/r/MachineLearning | 2021-10-10

  GitHub

• scikit-image

  3 4,546 9.5 Python

  Image processing in Python

  

  Project mention: Image Processing is Easier than you Thought! (Getting started with Python Pillow) | dev.to | 2021-10-13

  Python is a general-purpose programming language that provides many image processing libraries for adding image processing capabilities to digital images. Some of the most common image processing libraries in Python are OpenCV, Python Imaging Library (PIL), Scikit-image etc.

• RobustVideoMatting

  6 3,987 5.8 Python

  Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

  Project mention: [D] AI Background Removal: a quick comparison between RVM & BGMv2 | reddit.com/r/MachineLearning | 2021-10-10

  GitHub

• pysot

  1 3,843 0.4 Python

  SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.

  

  Project mention: [P] Object Detection with rotated bounding boxes | reddit.com/r/MachineLearning | 2021-01-25

  SiamMask has pretty good rotated bbox code you can take a look at. https://github.com/STVIR/pysot

• imagededup

  1 3,821 1.8 Python

  😎 Finding duplicate images made easy!

  

  Project mention: How to remove duplicate images from your dataset (Also CIFAR-100 has issues) | reddit.com/r/computervision | 2021-02-01

  I had used phash to find the duplicate images. The hashing algorithm is ingenious...And it was able to find duplicates (and thus remove) efficiently imagededub has perceptual hashing.

• pytorch-metric-learning

  1 3,704 8.5 Python

  The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

  Project mention: [D] Similar Image Retrieval | reddit.com/r/MachineLearning | 2021-06-23

  This repo provides the tools and examples needed to build such a model: https://github.com/KevinMusgrave/pytorch-metric-learning

• autogluon

  1 3,696 9.2 Python

  AutoGluon: AutoML for Text, Image, and Tabular Data

  

  Project mention: What will the data science job market be like in 5 years? | reddit.com/r/datascience | 2021-08-14

  Some AutoML is getting pretty good, AutoGluon is very solid for tabular data. That being said you still need to have your data in tabular format and deployment still requires some effort.

Index

Sponsored

Scout APM: A developer's best friend. Try free for 14-days

Scout APM uses tracing logic that ties bottlenecks to source code so you know the exact line of code causing performance issues and can get back to building a great product faster.

scoutapm.com