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)
APACHE-2.0 License
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
Pytorch framework for doing deep learning on point clouds.
A collection of computer vision pre-trained models.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classific...
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
[ECCV 2022] SimpleRecon: 3D Reconstruction Without 3D Convolutions
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
Run PyTorch LLMs locally on servers, desktop and mobile
[ECCV 2024] DoubleTake: Geometry Guided Depth Estimation
Research Framework for easy and efficient training of GANs based on Pytorch