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
123 forks
Language: Python
last commit: 3 months ago
Linked from 1 awesome list
augmentationdeep-learningfocal-lossimage-classificationimage-processingimage-segmentationjaccard-losskagglemachine-learningobject-detectionpipelinepythonpytorchsegmentationtest-time-augmentationtta
bharathgs/awesome-pytorch-listRelated 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,236 |
 pytorch/tnt | A collection of tools and utilities for building and training neural networks with PyTorch. | 1,669 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,044 |
 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 | 172 |
 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 | 547 |
 jacobgil/pytorch-pruning | This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. | 877 |
 lightning-universe/lightning-bolts | Provides a toolbox of components to extend PyTorch Lightning for deep learning research and production | 1,700 |
 blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorch | An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts | 65 |