Python Computer Vision

Open-source Python projects categorized as Computer Vision | Edit details

Top 23 Python Computer Vision Projects

  • GitHub repo Face Recognition

    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

  • GitHub repo pytorch-CycleGAN-and-pix2pix

    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

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

  • GitHub repo EasyOCR

    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 :)

  • GitHub repo facenet

    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.

  • GitHub repo jina

    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

  • GitHub repo d2l-en

    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.

  • GitHub repo datasets

    🤗 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

    Run Linux Software Faster and Safer than Linux with Unikernels.

  • GitHub repo vision

    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.

  • GitHub repo fashion-mnist

    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.

  • GitHub repo labelme

    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/

  • GitHub repo meshroom

    3D Reconstruction Software

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

    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.

  • GitHub repo Serpent.AI

    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.

  • GitHub repo vit-pytorch

    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

  • GitHub repo pix2pixHD

    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)

  • GitHub repo Kornia

    Open Source Differentiable Computer Vision Library

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

    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

  • GitHub repo scikit-image

    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.

  • GitHub repo RobustVideoMatting

    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

  • GitHub repo pysot

    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

  • GitHub repo imagededup

    😎 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.

  • GitHub repo pytorch-metric-learning

    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

  • GitHub repo autogluon

    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.

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2021-10-16.

Index

What are some of the best open-source Computer Vision projects in Python? This list will help you:

Project Stars
1 Face Recognition 41,739
2 pytorch-CycleGAN-and-pix2pix 15,969
3 EasyOCR 12,810
4 facenet 12,058
5 jina 11,559
6 d2l-en 11,202
7 datasets 10,108
8 vision 10,014
9 fashion-mnist 9,507
10 labelme 7,343
11 meshroom 6,997
12 label-studio 6,709
13 Serpent.AI 6,052
14 vit-pytorch 5,999
15 pix2pixHD 5,476
16 Kornia 5,063
17 BackgroundMattingV2 4,858
18 scikit-image 4,546
19 RobustVideoMatting 3,987
20 pysot 3,843
21 imagededup 3,821
22 pytorch-metric-learning 3,704
23 autogluon 3,696
Find remote jobs at our new job board 99remotejobs.com. There are 34 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
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