Hina Detect Model

模型/代码来源

该模型基于orchidsziyou的仓库SeiaDetectModel训练而来，其基于spawner1145的仓库NailongRecognizeRL，根据 GNU Affero General Public License (AGPL) 版本 3 进行发布。

模型介绍

该模型基于预训练的ResNet-50模型，使用了我个人收集的500+张 空崎日奈 的图片跟1k+张其他各种图片训练而来，可以检测出包含Hina的图。

原项目模型主要是为QQbot进行训练的,为了占用尽可能小的空间，将训练好的.pth模型文件转换为ONNX格式，并使用ONNXRuntime库进行推理。

模型使用方法

2025/5/23更新：基于本仓库的模型编写的astrbot的插件：
astrbot_plugin_seiadetect

环境准备

参考requirements.txt安装所需的依赖库

训练模型

仓库已经提供了训练好的模型，如果需要重新训练模型，可以参考train01.py文件进行训练。

在我自己训练的时候，由于正数据相对于负数据集太少，因此在训练之前使用flip_picture.py把正训练集当中的所有图片进行上下翻转来扩充数据集。

另外，由于数据集过大，而且可能包括一些别的照片，因此没有提供数据集，若想要自己训练，需要自己收集数据集。

预测

使用PredictONNX.py来预测图片是否包含Hina。 使用clipboard.py来预测剪贴板图片是否包含Hina。

注意事项

若不使用ONNXRuntime库进行推理，本仓库也提供了.pth模型文件，可以直接使用pytorch进行推理，推理代码参考run.py(代码来源@spawner1145)

存在问题

模型效果

可以认出穿圣娅泳装的hina pid:133158368

穿hina衣服的星野可以认出不是hina pid:132695434