LucaBonfiglioli / nnviz
Torch nn vizualization
☆51Updated last year
Alternatives and similar repositories for nnviz:
Users that are interested in nnviz are comparing it to the libraries listed below
- ☆66Updated 3 weeks ago
- Timm model explorer☆39Updated last year
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆573Updated last month
- Self-Supervised Learning in PyTorch☆136Updated last year
- Each week I create sketches covering key Computer Vision concepts. If you want to learn more about CV stick around!☆148Updated 2 years ago
- A high-performance library for detecting objects in images and videos, leveraging Rust's speed and safety. Optionally supports a gRPC API…☆30Updated this week
- A curated list of papers that released datasets along with their work☆125Updated 5 months ago
- Interactively inspect module inputs, outputs, parameters, and gradients.☆333Updated 3 months ago
- Large Scale Non maximum Suppression. This project implements a O(nlog(n)) approach for non max suppression, useful for object detection r…☆68Updated 6 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆114Updated last month
- A tool for converting computer vision label formats.☆62Updated this week
- PhyCV: The First Physics-inspired Computer Vision Library☆504Updated 5 months ago
- RobustSAM: Segment Anything Robustly on Degraded Images (CVPR 2024 Highlight)☆347Updated 7 months ago
- SAM Annotaton Tool☆33Updated 9 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆424Updated 4 months ago
- Cache PyTorch module outputs on-the-fly☆40Updated 7 months ago
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆111Updated last year
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆40Updated last month
- This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.☆168Updated 3 years ago
- Simplify Your Visual Data Ops. Find and visualize issues with your computer vision datasets such as duplicates, anomalies, data leakage, …☆68Updated last year
- torchview: visualize pytorch models☆905Updated last month
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆230Updated 2 months ago
- [ICML 2023] Official PyTorch implementation of Global Context Vision Transformers☆429Updated last year
- Hiera: A fast, powerful, and simple hierarchical vision transformer.☆967Updated last year
- Torchvision-like Deformable Convolution with both 1D, 2D, 3D operators, and their transposed versions.☆26Updated 2 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 8 months ago
- ☆65Updated 6 months ago
- vision transformers with pytorch and pytorch lightningUpdated 6 months ago
- Transform datasets at scale. Optimize datasets for fast AI model training.☆447Updated this week
- Use Segment Anything 2, grounded with Florence-2, to auto-label data for use in training vision models.☆119Updated 8 months ago