sithu3

semantic-segmentation. SOTA Semantic Segmentation Models in PyTorch

PyTorch-ONNX-TFLite. Conversion of PyTorch Models into TFLite

sota-backbones. A collection of SOTA Image Classification Models in PyTorch

pose-estimation. Easy to use SOTA Top-Down Multi-person Pose Estimation Models in PyTorch

EasyFace. Easy-to-use Face Analysis Tool

PyTorch-ONNX-TRT. TensortRT installation and Conversion from PyTorch Models

simple-object-tracking. Object Tracking with YOLOv5, CLIP, DINO and DeepSORT

LOD. List of Computer Vision Datasets

CUDA-Install-Guide. Installation guide for NVIDIA driver, CUDA, cuDNN and TensorRT

audio-tagging. Easy to use Audio Tagging in PyTorch

DGSfM. Depth-Guided Scale-Aware Global Structure-from-Motion

self-supervised-learning. Fast, Simple and Easy to use SOTA Self-Supervised Image Classification Models in PyTorch

image-captioning. Simple and Easy to use Image Captioning Implementation

torch_optimize. Optimize PyTorch Models for faster inference

video-classification. Simple pipeline to get started with video classification and action recognition

pysfm. Python only Structure-from-Motion and 3D Computer Vision (with SOTA models)

Knowledge-Distillation-Pipeline. PyTorch Knowledge Distillation Framework

easytrain. Reusable training tool

coco-caption. Jupyter Notebook

panoptic-segmentation. SOTA Panoptic Segmentation Models in PyTorch

magicnerf. A user-friendly and high-performance implementation of neural radiance fields (NeRF)