search

facebookresearch / detr

End-to-End Object Detection with Transformers
Apache License 2.0
13.42k stars 2.42k forks source link

max_size do not work in RandomResize() #574

Open ZhouZhuo2022 opened 1 year ago

ZhouZhuo2022 commented 1 year ago 


``` class RandomResize():
    def __init__(self,sizes,max_size = None):
        assert isinstance(sizes,(list,tuple))
        self.sizes = sizes
        self.max_size = max_size
    def __call__(self,img,target=None):
        size = random.choice(self.sizes)
        return resize(img,target,size,self.max_size)

IMG = RandomResize([[1400,1400],[1400,1400]],400)(image)
## in this condiction the max_size doesn't work