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

Velocity · 7d
+0.6
★ / day
Trend
→steady
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.