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
122 forks
Language: Python
last commit: about 1 month ago
Linked from 1 awesome list

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

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
asyml/texar-pytorch A toolkit providing easy-to-use machine learning modules and functionalities for natural language processing and text generation tasks 745
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,217
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,666
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
vermeille/torchelie A collection of utility functions and tools for building deep learning models with PyTorch 111
speedinghzl/pytorch-segmentation-toolbox Provides PyTorch implementations of deep learning models for semantic segmentation 772
dmarnerides/pydlt A PyTorch-based toolbox for building and training deep learning models with ease. 204
vinhkhuc/pytorch-mini-tutorials A collection of tutorials and lessons on building deep learning models using the PyTorch library. 326
zudi-lin/pytorch_connectomics A deep learning framework for automatic and semi-automatic segmentation of 3D image stacks in connectomics 171
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
ptrblck/pytorch_misc A collection of code snippets and examples exploring various PyTorch concepts and techniques 544
jacobgil/pytorch-pruning This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. 875
lightning-universe/lightning-bolts Provides a toolbox of components to extend PyTorch Lightning for deep learning research and production 1,693
blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorch An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts 65