alexsr

yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

torch-cam

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

obs-backgroundremoval

An OBS plugin for removing background in portrait images (video), making it easy to replace the background when recording or streaming.

willow

Open source, local, and self-hosted Amazon Echo/Google Home competitive Voice Assistant alternative

my_notes

My small cheatsheets for data science, ML, computer science and more.

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

deep-rl-class

This repo contains the syllabus of the Hugging Face Deep Reinforcement Learning Course.

evotorch

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

pyro

Deep universal probabilistic programming with Python and PyTorch

flowtorch

This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a community of users and contributors by focusing initially on complete infra and documentation for how to use and create components.

x-unet

Implementation of a U-net complete with efficient attention as well as the latest research findings

BayesCap

(ECCV 2022) BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networks

flash-attention

Fast and memory-efficient exact attention

composer

Supercharge Your Model Training

ML-YouTube-Courses

📺 Discover the latest machine learning / AI courses on YouTube.

Objectron

Objectron is a dataset of short, object-centric video clips. In addition, the videos also contain AR session metadata including camera poses, sparse point-clouds and planes. In each video, the camera moves around and above the object and captures it from different views. Each object is annotated with a 3D bounding box. The 3D bounding box describes the object’s position, orientation, and dimensions. The dataset contains about 15K annotated video clips and 4M annotated images in the following categories: bikes, books, bottles, cameras, cereal boxes, chairs, cups, laptops, and shoes

IA-SSD

Not All Points Are Equal: Learning Highly Efficient Point-based Detectors for 3D LiDAR Point Clouds (CVPR 2022, Oral)

portable-file-dialogs

💬 Portable GUI dialogs library, C++11, single-header

kubric

A data generation pipeline for creating semi-realistic synthetic multi-object videos with rich annotations such as instance segmentation masks, depth maps, and optical flow.

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

evojax

dm-haiku

JAX-based neural network library

projected-gan

[NeurIPS'21] Projected GANs Converge Faster

PVT

PVT: Point-Voxel Transformer for 3D Deep Learning

pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"

torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

ndf

Neural Unsigned Distance Fields (NDF) - Codebase

score_flow

Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)

ADOP