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!

vak

A neural network framework for researchers studying acoustic communication

torchlm

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

facetorch

Python library for analysing faces using PyTorch

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.

torchvision-enhance

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

betterloader

A better PyTorch dataloader

pytorch-cnn-trainer

A simple yet powerful CNN trainer for PyTorch and Lightning.

built

BuilT - Build a Trainer of deep neural networks

pytorch-fasterrcnn

Torchvision Faster RCNN Fine Tuner

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.

torch-featurelayer

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

artsnake

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

opencv-torchvision-transforms-yuzhiyang

opencv reimplement for transforms in torchvision

imread-benchmark

I/O benchmark for different image processing python libraries.

torchconvview

A library for PyTorch convolution layer visualizations.

torchsat

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

quadratum

Additional torchvision image transforms for practical usage.

inferflow

Universal Inference Pipeline Framework