Skip to content

Instantly share code, notes, and snippets.

View MohamedAliRashad's full-sized avatar
🎯
Focusing

Mohamed Rashad MohamedAliRashad

🎯
Focusing
View GitHub Profile
@MohamedAliRashad
MohamedAliRashad / spatial_pyramid_pooling_torch.py
Created November 14, 2022 01:32
Spatial Pyramid Pooling Pytorch
## Credits to https://github.com/revidee/pytorch-pyramid-pooling
## I just modified on it to make it simpler
class SpatialPyramidPooling(nn.Module):
def __init__(self, partition_size_list: list, pooling_mode: str = "max"):
"""Spatial Pyramid Pooling layer for PyTorch users
Parameters
----------
partition_size_list : list