mirror of
https://github.com/Refound-445/nonebot-plugin-nailongremove.git
synced 2025-11-04 21:22:43 +08:00
up
This commit is contained in:
parent
7999c41980
commit
bce2e7ab9e
@ -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="一个基于图像分类模型的简单插件~",
|
||||
|
||||
@ -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(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user