yjh0410 1 년 전
부모
커밋
c271a3bd35
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      dataset/data_augment/ssd_augment.py

+ 1 - 1
dataset/data_augment/ssd_augment.py

@@ -498,7 +498,7 @@ class SSDAugmentation(object):
 
         # ----------------- Transforms -----------------
         self.augment = Compose([
-            RandomDistort(prob=0.8),
+            RandomDistort(prob=0.5),
             RandomExpand(fill_value=self.pixel_mean[::-1]),
             RandomIoUCrop(p=0.8),
             RandomHorizontalFlip(p=0.5),