Torchvision Python Packages

torchinfo

View model summaries in PyTorch!

torchgeo

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

torchsr

Super Resolution datasets and models in Pytorch

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.

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.🎉

vak

A neural network framework for researchers studying acoustic communication

body-matrix

Calculate Body Measurements with Computer Vision

alicia

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

torchvision-detection

Object detection reference training scripts.

pytorch-cnn-trainer

A Simple but Powerful CNN Trainer For PyTorch

keras-aug

A library that includes Keras 3 preprocessing and augmentation layers

torchvision-enhance

🔥TorchSat 🌏 is an open-source deep learning framework for satellite imagery analysis based on PyTorch.

built

BuilT - Build a Trainer of deep neural networks

torch-featurelayer

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

betterloader

A better PyTorch dataloader

pytorch-fasterrcnn

Fine-Tune Pytorch Faster RCNN for your own task.

artsnake

A brief description of your library

opencv-torchvision-transforms-yuzhiyang

opencv reimplement for transforms in torchvision

imread-benchmark

I/O benchmark for different image processing python libraries.

inferflow

Universal Inference Pipeline Framework for Computer Vision

torchconvview

A library for PyTorch convolution layer visualizations.

torchsat

TorchSat is an open-source PyTorch framework for satellite imagery analysis.