Stars
FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Python sample codes and textbook for robotics algorithms.
Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Image-to-Image Translation in PyTorch
LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source …
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Code for the paper "Language Models are Unsupervised Multitask Learners"
A cross-platform command-line utility that creates projects from cookiecutters (project templates), e.g. Python package projects, C projects.
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Graph Neural Network Library for PyTorch
Open standard for machine learning interoperability
Tool for producing high quality forecasts for time series data that has multiple seasonality with linear or non-linear growth.
Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"
Faker is a Python package that generates fake data for you.
A Deep Learning based project for colorizing and restoring old images (and video!)
PyTorch implementations of Generative Adversarial Networks.
Datasets, Transforms and Models specific to Computer Vision
Machine Learning Engineering Open Book
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
End-to-End Object Detection with Transformers
Image augmentation for machine learning experiments.
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
StyleGAN - Official TensorFlow Implementation
Convert Machine Learning Code Between Frameworks
Python package built to ease deep learning on graph, on top of existing DL frameworks.
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
An open source implementation of CLIP.