Top Python Computer vision repositories on GitHub
Image and video understanding: detection, segmentation, generation. Filtered to projects whose primary language is Python.
Ranked by stars across 3,122 Python repositories tagged computer-vision. Refreshed daily.
- 1d2l-ai/d2l-zh★ 77,696 · ⑂ 12,281
《动手学深度学习》:面向中文读者、能运行、可讨论。中英文版被70多个国家的500多所大学用于教学。
- deep-learning
- book
- notebook
- natural-language-processing
- computer-vision
- machine-learning
- 2ultralytics/ultralytics★ 56,820 · ⑂ 10,921
Ultralytics YOLO 🚀
- hub
- ultralytics
- yolov8
- yolo
- deep-learning
- image-classification
- 3roboflow/supervision★ 38,361 · ⑂ 3,414
We write your reusable computer vision tools. 💜
- computer-vision
- image-processing
- python
- yolo
- instance-segmentation
- object-detection
- 4d2l-ai/d2l-en★ 28,777 · ⑂ 5,055
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
- deep-learning
- machine-learning
- book
- notebook
- computer-vision
- natural-language-processing
- 5lucidrains/vit-pytorch★ 25,149 · ⑂ 3,495
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
- artificial-intelligence
- attention-mechanism
- transformers
- computer-vision
- image-classification
- 6junyanz/pytorch-CycleGAN-and-pix2pix★ 25,105 · ⑂ 6,576
Image-to-Image Translation in PyTorch
- pytorch
- gan
- cyclegan
- pix2pix
- deep-learning
- computer-vision
- 7graphdeco-inria/gaussian-splatting★ 21,678 · ⑂ 3,166
Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"
- computer-graphics
- computer-vision
- radiance-field
- 8huggingface/datasets★ 21,492 · ⑂ 3,196
🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
- nlp
- datasets
- pytorch
- tensorflow
- pandas
- numpy
- 9pytorch/vision★ 17,664 · ⑂ 7,229
Datasets, Transforms and Models specific to Computer Vision
- computer-vision
- machine-learning
- 10wkentaro/labelme★ 15,823 · ⑂ 3,671
Image annotation with Python. Supports polygon, rectangle, circle, line, point, and AI-assisted annotation.
- image-annotation
- python
- computer-vision
- annotations
- deep-learning
- semantic-segmentation
- 11cvat-ai/cvat★ 15,778 · ⑂ 3,667
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
- video-annotation
- computer-vision
- computer-vision-annotation
- deep-learning
- image-annotation
- annotation-tool
- 12davidsandberg/facenet★ 14,320 · ⑂ 4,785
Face recognition using Tensorflow
- face-recognition
- tensorflow
- facenet
- deep-learning
- computer-vision
- face-detection
- 13mlfoundations/open_clip★ 13,773 · ⑂ 1,281
An open source implementation of CLIP.
- deep-learning
- pytorch
- computer-vision
- language-model
- multi-modal-learning
- contrastive-loss
- 14jacobgil/pytorch-grad-cam★ 12,793 · ⑂ 1,702
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
- deep-learning
- pytorch
- grad-cam
- visualizations
- interpretability
- interpretable-ai
- 15zalandoresearch/fashion-mnist★ 12,725 · ⑂ 3,075
A MNIST-like fashion product database. Benchmark :point_down:
- mnist
- deep-learning
- benchmark
- machine-learning
- dataset
- computer-vision
- 16ludwig-ai/ludwig★ 11,688 · ⑂ 1,217
Low-code framework for building custom LLMs, neural networks, and other AI models
- deep-learning
- deeplearning
- deep
- learning
- machine-learning
- machinelearning
- 17qubvel-org/segmentation_models.pytorch★ 11,532 · ⑂ 1,835
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
- segmentation
- image-processing
- pspnet
- unet
- unet-pytorch
- pytorch
- 18nerfstudio-project/nerfstudio★ 11,525 · ⑂ 1,620
A collaboration friendly studio for NeRFs
- nerf
- pytorch
- 3d
- 3d-graphics
- 3d-reconstruction
- computer-vision
- 19kornia/kornia★ 11,196 · ⑂ 1,180
🐍 Geometric Computer Vision Library for Spatial AI
- computer-vision
- image-processing
- machine-learning
- pytorch
- deep-learning
- neural-network
- 20voxel51/fiftyone★ 10,638 · ⑂ 751
Refine high-quality datasets and visual AI models
- machine-learning
- artificial-intelligence
- deep-learning
- computer-vision
- developer-tools
- data-science
- 21autogluon/autogluon★ 10,290 · ⑂ 1,141
Fast and Accurate ML in 3 Lines of Code
- automl
- machine-learning
- data-science
- deep-learning
- ensemble-learning
- computer-vision
- 22xuebinqin/U-2-Net★ 9,722 · ⑂ 1,610
The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
- computer-vision
- deep-learning
- image-background-removal
- image-processing
- image-segmentation
- u-2-net
- 23PeterL1n/RobustVideoMatting★ 9,333 · ⑂ 1,196
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
- computer-vision
- machine-learning
- ai
- matting
- deep-learning
- 24CVHub520/X-AnyLabeling★ 9,000 · ⑂ 976
Effortless data labeling with AI support from Segment Anything and other awesome models.
- sam
- yolo
- deep-learning
- clip
- object-detection
- pose-estimation
- 25PeterL1n/BackgroundMattingV2★ 7,157 · ⑂ 963
Real-Time High-Resolution Background Matting
- computer-vision
- machine-learning
- matting
- real-time
Find Python engineers shipping Computer vision
The list above ranks the most-starred public Python repositories tagged with the Computer vision topic, drawn from the public GitHub graph. Across 3,122 matching repositories, the contributors are a tight cluster of engineers with both Python chops and real Computer vision experience.
That overlap is rare. Most Python engineers haven’t shipped Computer vision, and most Computer vision maintainers don’t write Python. The people on this list’s contributor graph are the ones who do both.
Refolk turns this list into a search. Ask for “Python Computer vision maintainers hiring” or “Python engineers shipping Computer vision in 2025” and Refolk returns a ranked shortlist with the commits, profiles, and projects behind each name.
How this list is built
Last refreshed: Thu, 07 May 2026 05:54:22 GMT
Need a more specific search?
Refolk runs natural-language searches across GitHub, LinkedIn, and the open web. Try one of these:
Related lists
- Python · Machine learning
- Python · Deep learning
- Python · Natural language processing
- Python · LLM
- Python · AI agents
- Python · RAG
- Python · Embeddings
- Python · Transformers
See all repository lists.