hualubei2020-callingsmoking icon indicating copy to clipboard operation
hualubei2020-callingsmoking copied to clipboard

數據預處理

Open ccqedq opened this issue 3 years ago • 3 comments
trafficstars

在train.py 裏直接加載的預處理後的數據,但預處理直接按最大邊等比縮放到600,對於不同長寬比的數據這樣處理不太合適吧,感覺padding應該更好一點

ccqedq avatar May 10 '22 03:05 ccqedq

数据增强里有padding https://github.com/fire717/hualubei2020-callingsmoking/blob/0d0f804cef38bea45a7063190f17d031008bf9c9/libs/data.py#L57-L64

fire717 avatar May 10 '22 03:05 fire717

min_size = max(img.shape[:2])

img = A.OneOf([A.PadIfNeeded(min_height=min_size, min_width=min_size, border_mode=3, value=0, mask_value=0, always_apply=False, p=0.7), A.PadIfNeeded(min_height=min_size, min_width=min_size, border_mode=0, value=0, mask_value=0, always_apply=False, p=0.3)], p=1.0)(image=img)['image'] 這裏不應該padding到self.h,self.w嗎

ccqedq avatar May 10 '22 03:05 ccqedq

假如你输入的是1000x2000的图 不先按最大边等比缩放到300x600再padding 你怎么直接padding到600x600? 这些问题你把每一步图片保存下来看看就知道了

fire717 avatar May 10 '22 05:05 fire717