Torchvision Python Packages

torchinfo

View model summaries in PyTorch!

torchsr

Super Resolution datasets and models in Pytorch

torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

torch-summary

View model summaries in PyTorch!

bnnr

XAI-driven augmentation & diagnostics for PyTorch vision - find model failures, fix with saliency-guided augmentation (ICD/AICD), prove with auditable reports.

body-matrix

Calculate Body Measurements with Computer Vision

facetorch

Python library for analysing faces using PyTorch

torchlm

💎An easy-to-use PyTorch library for face landmarks detection: training, evaluation, inference, and 100+ data augmentations.🎉

priml

Strongly typed configurable building blocks for A/B testing ML research.

vak

A neural network framework for researchers studying acoustic communication

segcraft

A config-first toolkit for semantic segmentation training, evaluation, and image/video inference.

alicia

A CLI to download, create, modify, train, test, predict and compare an image classifiers.

betterloader

A better PyTorch data loader capable of custom image operations and image subsets

built

BuilT - Build a Trainer of deep neural networks

artsnake

Library for watermarking and poisoning of images, created in an effort to stop digital art theft from fraudsters and AI scrapers

vision3d

A 3D extension of torchvision.

keras-aug

A library that includes Keras 3 preprocessing and augmentation layers, providing support for various data types such as images, labels, bounding boxes, segmentation masks, and more.

inferflow

Universal Inference Pipeline Framework

imread-benchmark

I/O benchmark for different image processing python libraries.

pytorch-cnn-trainer

A Simple but Powerful CNN Trainer For PyTorch

torch-dimensions

Torch for Multidimensional Compatibility and Easy to Go protocols for both CUDA and MPS. (SSMs are also provided probes for Apple MPS training/inferencing)

quadratum

Additional torchvision image transforms for practical usage.

torch-featurelayer

🧠 Useful utility functions and wrappers for hooking onto layers within PyTorch models for feature extraction.

pytorch-fasterrcnn

Fine-Tune Pytorch Faster RCNN for your own task.