hualubei2020-callingsmoking
hualubei2020-callingsmoking copied to clipboard
數據預處理
在train.py 裏直接加載的預處理後的數據,但預處理直接按最大邊等比縮放到600,對於不同長寬比的數據這樣處理不太合適吧,感覺padding應該更好一點
数据增强里有padding https://github.com/fire717/hualubei2020-callingsmoking/blob/0d0f804cef38bea45a7063190f17d031008bf9c9/libs/data.py#L57-L64
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嗎
假如你输入的是1000x2000的图 不先按最大边等比缩放到300x600再padding 你怎么直接padding到600x600? 这些问题你把每一步图片保存下来看看就知道了