From bce2e7ab9ecf4a7b3b3d3909c3f2345699453904 Mon Sep 17 00:00:00 2001 From: student_2333 Date: Thu, 31 Oct 2024 21:43:17 +0800 Subject: [PATCH] up --- nonebot_plugin_nailongremove/__init__.py | 2 +- nonebot_plugin_nailongremove/model/target_detection.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nonebot_plugin_nailongremove/__init__.py b/nonebot_plugin_nailongremove/__init__.py index 5da73b3..b78597b 100644 --- a/nonebot_plugin_nailongremove/__init__.py +++ b/nonebot_plugin_nailongremove/__init__.py @@ -9,7 +9,7 @@ require("nonebot_plugin_uninfo") from . import handler as handler from .config import Config -__version__ = "2.2.0" +__version__ = "2.2.0.post1" __plugin_meta__ = PluginMetadata( name="自动撤回奶龙", description="一个基于图像分类模型的简单插件~", diff --git a/nonebot_plugin_nailongremove/model/target_detection.py b/nonebot_plugin_nailongremove/model/target_detection.py index 86b857a..bed4168 100644 --- a/nonebot_plugin_nailongremove/model/target_detection.py +++ b/nonebot_plugin_nailongremove/model/target_detection.py @@ -80,7 +80,7 @@ def check_image(image: np.ndarray) -> "CheckResult": has = any( True for c, s in zip(final_cls_inds, final_scores) - if labels[c] == "nailong" and s >= config.nailong_model1_score + if labels[int(c)] == "nailong" and s >= config.nailong_model1_score ) if has: image = vis(