Repository Index

Tensors and Dynamic neural networks in Python with strong GPU acceleration

• neural-network
• autograd
• gpu
• numpy

We write your reusable computer vision tools. 💜

• computer-vision
• image-processing
• python
• yolo

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

• deep-learning
• pytorch
• gpu
• machine-learning

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

• ocr
• deep-learning
• crnn
• pytorch

State-of-the-art 2D and 3D Face Analysis Project

• face-recognition
• face-detection
• mxnet
• face-alignment

deep learning for image processing including classification and object-detection etc.

• pytorch
• tensorflow2
• classification
• object-detection

Image-to-Image Translation in PyTorch

• pytorch
• gan
• cyclegan
• pix2pix

pix2tex: Using a ViT to convert images of equations into LaTeX code.

• machine-learning
• transformer
• im2latex
• deep-learning

A collaboration friendly studio for NeRFs

• nerf
• pytorch
• 3d
• 3d-graphics

🐍 Geometric Computer Vision Library for Spatial AI

• computer-vision
• image-processing
• machine-learning
• pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

• segmentation
• image-processing
• pspnet
• unet

Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/ (no longer working)

• manga
• anime
• ocr
• deep-learning

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

OpenMMLab's next-generation platform for general 3D object detection.

• pytorch
• 3d-object-detection
• object-detection
• point-cloud

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

• point-cloud
• 3d-detection
• autonomous-driving
• pv-rcnn

A PyTorch Library for Accelerating 3D Deep Learning Research

• pytorch
• 3d-deep-learning
• neural-networks
• differentiable-rendering

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

World's first general purpose 3D object detection codebse.

• nuscenes
• 3d-object-detection
• point-cloud
• object-detection

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The PyTorch implementation)

• 3d-object-detection
• lidar-point-cloud
• bevmap
• center

A 3D vision library from 2D keypoints: monocular and stereo 3D detection for humans, social distancing, and body orientation.

• 3d-object-detection
• 3d-detection
• 3d-deep-learning
• pytorch

Panoptic segmentation algorithms for 2D and 3D electron microscopy images

• deep-learning
• semantic-segmentation
• segmentation
• pytorch