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