优化按类别筛选
This commit is contained in:
parent
49f38b09ab
commit
796fb6bcad
2
AI.py
2
AI.py
|
|
@ -68,7 +68,7 @@ def score_filter_byClass(pdetections,score_para_2nd):
|
||||||
elif str(int(cls)) in score_para_2nd.keys():
|
elif str(int(cls)) in score_para_2nd.keys():
|
||||||
score_th = score_para_2nd[str(int(cls))]
|
score_th = score_para_2nd[str(int(cls))]
|
||||||
else:
|
else:
|
||||||
score_th = 0.7
|
score_th = 0.25
|
||||||
if score > score_th:
|
if score > score_th:
|
||||||
ret.append(det)
|
ret.append(det)
|
||||||
return ret
|
return ret
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue