Pytorch pad to square

import torchvision.transforms.functional as F

class SquarePad:
    def __call__(self, image):
        max_wh = max(image.size)
        p_left, p_top = [(max_wh - s) // 2 for s in image.size]
        p_right, p_bottom = [max_wh - (s+pad) for s, pad in zip(image.size, [p_left, p_top])]
        padding = (p_left, p_top, p_right, p_bottom)
        return F.pad(image, padding, 0, 'constant')

target_image_size = (224, 224)  # as an example
# now use it as the replacement of transforms.Pad class
transform=transforms.Compose([
    SquarePad(),
    transforms.Resize(target_image_size),
    transforms.ToTensor(),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
])

Source: discuss.pytorch.org

Similar codes

9 votes169 views3 answers
SettingWithCopyWarning: A value is trying to be set on a copy of a slice from a DataFrame. Try using .loc[row_indexer,col_indexer] = value instead
Python
3 votes262 views5 answers
Python sha256 crypt decrypt
Python
0 vote166 views1 answer
Django file upload this field is required
Python
7 votes203 views3 answers
Spacy en_core_web_sm error
Python
1 vote223 views1 answer
How to count post by category django
Python Django
47 votes185 views3 answers
Convert to string in python
Python
1 vote194 views1 answer
Generate valid sudoku board python
Python
2 votes167 views1 answer
Binomial coefficient python
Python
0 vote167 views1 answer
Jupyter notebook don't show plot
Python
0 vote179 views1 answer
Django dump previous migrations
Python