← all repositories

BloodAxe/pytorch-toolbelt

A PyTorch extension library providing encoder-decoder building blocks, custom neural network modules, and GPU-friendly utilities for computer vision research.

1.6k stars Python ML FrameworksComputer Vision
pytorch-toolbelt
Not currently ranked — collecting fresh signals.
star history

This library offers a collection of PyTorch extensions for fast deep learning research and Kaggle competition workflows. It includes reusable encoder-decoder architectures, computer vision modules like CoordConv and SCSE, focal and Jaccard losses, and GPU-accelerated test-time augmentation for image segmentation and classification tasks.

Frequently asked

What is BloodAxe/pytorch-toolbelt?
A PyTorch extension library providing encoder-decoder building blocks, custom neural network modules, and GPU-friendly utilities for computer vision research.
Is pytorch-toolbelt open source?
Yes — BloodAxe/pytorch-toolbelt is open source, released under the MIT license.
What language is pytorch-toolbelt written in?
BloodAxe/pytorch-toolbelt is primarily written in Python.
How popular is pytorch-toolbelt?
BloodAxe/pytorch-toolbelt has 1.6k stars on GitHub.
Where can I find pytorch-toolbelt?
BloodAxe/pytorch-toolbelt is on GitHub at https://github.com/BloodAxe/pytorch-toolbelt.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.