fix: threshold of output

deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml

@@ -29,7 +29,7 @@ PreProcess:
 PostProcess:
   main_indicator: ThreshOutput
   ThreshOutput:
-    threshold: 0.9
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
   SavePreLabel:

ppcls/configs/PULC/safety_helmet/MobileNetV3_large_x0_35.yaml → ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml

@@ -16,7 +16,7 @@ Global:
 
 # model architecture
 Arch:
-  name: MobileNetV3_large_x0_35
+  name: MobileNetV3_small_x0_35
   pretrained: True
   class_num: 2
  
@@ -36,7 +36,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 0.050
+    learning_rate: 0.015
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -67,7 +67,7 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 256
+      batch_size: 64
       drop_last: False
       shuffle: True
     loader:
@@ -120,16 +120,16 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
 Metric:
   Train:
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]
   Eval:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0.yaml

@@ -119,16 +119,16 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
 Metric:
   Train:
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]
   Eval:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_distillation.yaml

@@ -169,7 +169,7 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
@@ -177,9 +177,9 @@ Metric:
   Train:
     - DistillationTopkAcc:
         model_key: "Student"
-        topk: [1, 2]
+        topk: [1]
   Eval:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_search.yaml

@@ -133,16 +133,16 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
 Metric:
   Train:
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]
   Eval:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]

ppcls/configs/PULC/safety_helmet/Res2Net200_vd_26w_4s.yaml

@@ -121,16 +121,17 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
 Metric:
   Train:
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]
   Eval:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]
+

ppcls/configs/PULC/safety_helmet/SwinTransformer_tiny_patch4_window7_224.yaml

@@ -139,7 +139,7 @@ Infer:
     - ToCHWImage:
   PostProcess:
     name: ThreshOutput
-    threshold: 0.9235
+    threshold: 0.5
     label_0: wearing_helmet
     label_1: unwearing_helmet
 
@@ -148,4 +148,4 @@ Metric:
     - TprAtFpr:
         max_fpr: 0.0001
     - TopkAcc:
-        topk: [1, 2]
+        topk: [1]