Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

Pytorch implementation of PCT: Point Cloud Transformer

Code for PointAugment: an Auto-Augmentation Framework for Point Cloud Classification, CVPR 2020 (Oral)

Official Tensorflow implementation of "Symmetrical Synthesis for Deep Metric Learning" (AAAI 2020)

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).

Open source deep learning based unsupervised image retrieval toolbox built on PyTorch🔥

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

We propose UniPose, a unified framework for human pose estimation, based on our “Waterfall” Atrous Spatial Pooling architecture, that achieves state-of-art-results on several pose estimation metric…

Chamfer Distance in Pytorch with f-score

Toolbox of models, callbacks, and datasets for AI/ML researchers.

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 came…

Differentiable Point Cloud Sampling (CVPR 2020 Oral)

The Replica Dataset v1 as published in https://arxiv.org/abs/1906.05797 .

[CVPR'19] Dataset and code used in the research project Scan2CAD: Learning CAD Model Alignment in RGB-D Scans

Deezer source separation library including pretrained models.

FMA: A Dataset For Music Analysis

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

PyTorch ,ONNX and TensorRT implementation of YOLOv4

🔎 Super-scale your images and run experiments with Residual Dense and Adversarial Networks.

An extension of Open3D to address 3D Machine Learning tasks

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

The PyTorch implementation of ICCV 2019 oral paper: free-form inpainting (deepfillv2), especially Gated Conv

Pytorch framework for doing deep learning on point clouds.

SynthDet - An end-to-end object detection pipeline using synthetic data

Keras implementation of RetinaNet object detection.

CVPR2020 PF-Net: Point Fractal Network for 3D Point Cloud Completion