Update modnet_mobilenetv2.yml

JC0624 · Dec 6, 2021 · b9a0540 · b9a0540
1 parent 9a47e58
commit b9a0540
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/contrib/Matting/configs/modnet/modnet_mobilenetv2.yml b/contrib/Matting/configs/modnet/modnet_mobilenetv2.yml
@@ -7,6 +7,8 @@ train_dataset:
   train_file: train.txt
   transforms:
     - type: LoadImages
+    - type: ResizeByShort
+      short_size: 512
     - type: RandomCrop
       crop_size: [512, 512]
     - type: RandomDistort