Procházet zdrojové kódy

fix cls train config

zhangyubo0722 před 10 měsíci
rodič
revize
7675ae7389

+ 5 - 3
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-L.yaml

@@ -139,9 +139,11 @@ Infer:
     - DecodeImage:
         to_rgb: True
         channel_first: False
-    - ResizeImage:
-        resize_short: 256
-    - CropImage:
+    - ResizeImage:  
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
+    - CropImage: 
         size: 224
     - NormalizeImage:
         scale: 1.0/255.0

+ 5 - 3
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-M.yaml

@@ -139,9 +139,11 @@ Infer:
     - DecodeImage:
         to_rgb: True
         channel_first: False
-    - ResizeImage:
-        resize_short: 256
-    - CropImage:
+    - ResizeImage:  
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
+    - CropImage: 
         size: 224
     - NormalizeImage:
         scale: 1.0/255.0

+ 4 - 2
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-S.yaml

@@ -140,8 +140,10 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 256
-    - CropImage:
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
+    - CropImage: 
         size: 224
     - NormalizeImage:
         scale: 1.0/255.0

+ 5 - 3
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-T0.yaml

@@ -137,9 +137,11 @@ Infer:
     - DecodeImage:
         to_rgb: True
         channel_first: False
-    - ResizeImage:
-        resize_short: 256
-    - CropImage:
+    - ResizeImage: 
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
+    - CropImage: 
         size: 224
     - NormalizeImage:
         scale: 1.0/255.0

+ 4 - 2
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-T1.yaml

@@ -139,8 +139,10 @@ Infer:
     - DecodeImage:
         to_rgb: True
         channel_first: False
-    - ResizeImage:
-        resize_short: 256
+    - ResizeImage:  
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
     - CropImage:
         size: 224
     - NormalizeImage:

+ 5 - 3
paddlex/repo_apis/PaddleClas_api/configs/FasterNet-T2.yaml

@@ -138,9 +138,11 @@ Infer:
     - DecodeImage:
         to_rgb: True
         channel_first: False
-    - ResizeImage:
-        resize_short: 256
-    - CropImage:
+    - ResizeImage:  
+        interpolation: bicubic
+        backend: pil
+        resize_short: 248
+    - CropImage: 
         size: 224
     - NormalizeImage:
         scale: 1.0/255.0

+ 4 - 2
paddlex/repo_apis/PaddleClas_api/configs/MobileNetV4_conv_large.yaml

@@ -157,9 +157,11 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 256
+        backend: pil
+        interpolation: bicubic
+        resize_short: 448
     - CropImage:
-        size: 224
+        size: 448
     - NormalizeImage:
         scale: 1.0/255.0
         mean: [0.485, 0.456, 0.406]

+ 4 - 2
paddlex/repo_apis/PaddleClas_api/configs/MobileNetV4_conv_medium.yaml

@@ -157,9 +157,11 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 256
+        backend: pil
+        interpolation: bicubic
+        resize_short: 320
     - CropImage:
-        size: 224
+        size: 320
     - NormalizeImage:
         scale: 1.0/255.0
         mean: [0.485, 0.456, 0.406]

+ 3 - 1
paddlex/repo_apis/PaddleClas_api/configs/MobileNetV4_conv_small.yaml

@@ -157,9 +157,11 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
+        backend: pil
+        interpolation: bicubic
         resize_short: 256
     - CropImage:
-        size: 224
+        size: 256
     - NormalizeImage:
         scale: 1.0/255.0
         mean: [0.485, 0.456, 0.406]

+ 2 - 0
paddlex/repo_apis/PaddleClas_api/configs/MobileNetV4_hybrid_large.yaml

@@ -157,6 +157,8 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
+        backend: pil
+        interpolation: bicubic
         resize_short: 448
     - CropImage:
         size: 448

+ 2 - 0
paddlex/repo_apis/PaddleClas_api/configs/MobileNetV4_hybrid_medium.yaml

@@ -152,6 +152,8 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
+        backend: pil
+        interpolation: bicubic
         resize_short: 256
     - CropImage:
         size: 256

+ 3 - 1
paddlex/repo_apis/PaddleClas_api/configs/StarNet-S1.yaml

@@ -142,7 +142,9 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 224
+        interpolation: bicubic
+        backend: pil
+        resize_short: 256
     - CropImage:
         size: 224
     - NormalizeImage:

+ 3 - 1
paddlex/repo_apis/PaddleClas_api/configs/StarNet-S2.yaml

@@ -142,7 +142,9 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 224
+        interpolation: bicubic
+        backend: pil
+        resize_short: 256
     - CropImage:
         size: 224
     - NormalizeImage:

+ 3 - 1
paddlex/repo_apis/PaddleClas_api/configs/StarNet-S3.yaml

@@ -142,7 +142,9 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 224
+        interpolation: bicubic
+        backend: pil
+        resize_short: 256
     - CropImage:
         size: 224
     - NormalizeImage:

+ 3 - 1
paddlex/repo_apis/PaddleClas_api/configs/StarNet-S4.yaml

@@ -142,7 +142,9 @@ Infer:
         to_rgb: True
         channel_first: False
     - ResizeImage:
-        resize_short: 224
+        interpolation: bicubic
+        backend: pil
+        resize_short: 256
     - CropImage:
         size: 224
     - NormalizeImage: