search

NVlabs / mask-auto-labeler

Other
159 stars 14 forks source link

some parameters for MAL #18

Open liusurufeng opened 1 year ago

liusurufeng commented 1 year ago

@voidrank hi,I'm sorry to bother you once again! There was a trouble when I tried to use my own dataset to train the MAL . in the file of data_aug.py,the line of 224: the code as follow: class RandomCropv1: def init(self, crop_sizes=[448], crop_fields=['image', 'mask']): self._crop_sizes = crop_sizes

How can I set the size of crop_sizes in my own size? the input size of my dataset is 256.

voidrank commented 1 year ago

Hi @liusurufeng as long as it is multiply of 32, it should be fine