diff --git a/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml b/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml index 5f128e187..66a4cebb3 100644 --- a/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml +++ b/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: diff --git a/ppcls/configs/PULC/safety_helmet/MobileNetV3_large_x0_35.yaml b/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml similarity index 95% rename from ppcls/configs/PULC/safety_helmet/MobileNetV3_large_x0_35.yaml rename to ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml index 3f945e49d..fd1817109 100644 --- a/ppcls/configs/PULC/safety_helmet/MobileNetV3_large_x0_35.yaml +++ b/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] diff --git a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0.yaml b/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0.yaml index bbc3c1414..3ad59e2e0 100644 --- a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0.yaml +++ b/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] diff --git a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_distillation.yaml b/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_distillation.yaml index 7d47ef079..254db5df4 100644 --- a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_distillation.yaml +++ b/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] diff --git a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_search.yaml b/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_search.yaml index d5f8b4553..58cc3300d 100644 --- a/ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_search.yaml +++ b/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] diff --git a/ppcls/configs/PULC/safety_helmet/Res2Net200_vd_26w_4s.yaml b/ppcls/configs/PULC/safety_helmet/Res2Net200_vd_26w_4s.yaml index b9660d6f0..5ab2f484e 100644 --- a/ppcls/configs/PULC/safety_helmet/Res2Net200_vd_26w_4s.yaml +++ b/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] + diff --git a/ppcls/configs/PULC/safety_helmet/SwinTransformer_tiny_patch4_window7_224.yaml b/ppcls/configs/PULC/safety_helmet/SwinTransformer_tiny_patch4_window7_224.yaml index dcc537347..666ad6be2 100644 --- a/ppcls/configs/PULC/safety_helmet/SwinTransformer_tiny_patch4_window7_224.yaml +++ b/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]