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

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