未验证 提交 08e27331 编写于 作者: G gaotingquan

fix: threshold of output

上级 ebe19ce8
......@@ -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:
......
......@@ -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]
......@@ -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]
......@@ -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]
......@@ -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]
......@@ -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]
......@@ -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]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册