Search 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!

facetorch

Python library for analysing faces using 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.🎉

body-matrix

Calculate Body Measurements with Computer Vision

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

pytorch-cnn-trainer

A simple yet powerful CNN trainer for PyTorch and Lightning.

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.

torchvision-enhance

Enhance torchvision for multi-channel images, 16-bit image, segmentation...

torchvision-detection

Object detection reference training scripts.

artsnake

A brief description of your library

pytorch-fasterrcnn

Fine-Tune Pytorch Faster RCNN for your own task.

torch-featurelayer

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

quadratum

Additional torchvision image transforms for practical usage.

inferflow

Universal Inference Pipeline Framework

opencv-torchvision-transforms-yuzhiyang

opencv reimplement for transforms in torchvision

torchconvview

A library for PyTorch convolution layer visualizations.

torchsat

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

gan-evaluator

A pip-installable evaluator for GANs (IS and FID). Accepts either dataloaders or individual batches. Supports on-the-fly evaluation during training. A working DCGAN SVHN demo script provided.