pytorch-toolbelt

ML toolkit

A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development.

PyTorch extensions for fast R&D prototyping and Kaggle farming

GitHub

2k stars
29 watching
123 forks
Language: Python
last commit: almost 2 years ago
Linked from 1 awesome list

augmentationdeep-learningfocal-lossimage-classificationimage-processingimage-segmentationjaccard-losskagglemachine-learningobject-detectionpipelinepythonpytorchsegmentationtest-time-augmentationtta

Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
asyml/texar-pytorchA toolkit providing easy-to-use machine learning modules and functionalities for natural language processing and text generation tasks745
kaiyangzhou/dassl.pytorchA PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.1,236
pytorch/tntA collection of tools and utilities for building and training neural networks with PyTorch.1,669
zhanghang1989/pytorch-encodingA Python framework for building deep learning models with optimized encoding layers and batch normalization.2,044
vermeille/torchelieA collection of utility functions and tools for building deep learning models with PyTorch111
speedinghzl/pytorch-segmentation-toolboxProvides PyTorch implementations of deep learning models for semantic segmentation772
dmarnerides/pydltA PyTorch-based toolbox for building and training deep learning models with ease.204
vinhkhuc/pytorch-mini-tutorialsA collection of tutorials and lessons on building deep learning models using the PyTorch library.326
zudi-lin/pytorch_connectomicsA deep learning framework for automatic and semi-automatic segmentation of 3D image stacks in connectomics172
pytorchbearer/torchbearerA PyTorch model fitting library designed to simplify the process of training deep learning models.636
nearai/torchfoldA PyTorch module for dynamic batching and optimized computation on deep neural networks221
ptrblck/pytorch_miscA collection of code snippets and examples exploring various PyTorch concepts and techniques547
jacobgil/pytorch-pruningThis project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference.877
lightning-universe/lightning-boltsProvides a toolbox of components to extend PyTorch Lightning for deep learning research and production1,700
blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorchAn implementation of cascaded refinement networks for generating photorealistic images from semantic layouts65