This commit is contained in:
student_2333 2024-10-31 19:52:34 +08:00
parent e36146ace5
commit f64105bd3e
No known key found for this signature in database
GPG Key ID: 665F083BEC56F2A6

View File

@ -77,7 +77,7 @@ def check_image(image: np.ndarray) -> "CheckResult":
has = any(
True
for c, s in zip(final_cls_inds, final_scores)
if c == 1 and s >= config.nailong_model1_score
if labels[c] == "nailong" and s >= config.nailong_model1_score
)
if has:
image = vis(