search

isht7 / pytorch-deeplab-resnet

DeepLab resnet v2 model in pytorch
MIT License
602 stars 118 forks source link

Random cropping during preprocessing in train.py #20

Open mgarbade opened 6 years ago

mgarbade commented 6 years ago

Thanks a lot for your repository. While reading through train.py I noticed that PASCAL VOC images are being resized to 321x321 in this line. In the original paper they perform a random cropping on img and gt mask. Here is an example from the corresponding Tensorflow code. There mask and image are concatenated and then cropped simultaneously.

isht7 commented 6 years ago

Thank you for your feedback! I will try to add this in some time.