Open3D: A Modern Library for 3D Data Processing

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

🤖💬 Transformer TTS: Implementation of a non-autoregressive Transformer based neural network for text to speech.

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

Python script to upload videos on YouTube using Selenium

Rembg is a tool to remove images background

A simple, cross-platform GUI automation module for Python and Rust.

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Turn (almost) any Python command line program into a full GUI application with one line

PyTorch extensions for fast R&D prototyping and Kaggle farming

NVIDIA's Deep Imagination Team's PyTorch Library

Motion Retargeting Video Subjects

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

A code-completion engine for Vim

gRPC server for hosting ML models trained on any framework in python

torch-optimizer -- collection of optimizers for Pytorch

On the Variance of the Adaptive Learning Rate and Beyond

Generating Digital Painting Lighting Effects via RGB-space Geometry (SIGGRAPH2020/TOG2020)

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Semantic Image Synthesis with SPADE

Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method.

End-to-End Object Detection with Transformers

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Background Matting: The World is Your Green Screen