Machine learning related utils includes implementations of:
Preprocessing
- GaussianFocus preprocessing (@pytorch) - also known as vignette
Modules
- Spatial Pyramid Pooling (@pytorch)
- Spatial Transform Network (@pytorch)
LrSched
- CosineAnnealRestartLRScheduler (@pytorch) - custom control of bottom
Loss
- CustomFocalLoss (@pytorch)