素材来源

icon 来自企鹅物流前端, 运行 images/download-icons.sh 下载

item 来自明日方舟工具箱, 运行 images/download-items.sh 下载

collect 来自yuanyan3060/Arknights-Bot-Resource, 运行 dl_data.py 下载

训练

运行 train_torch_from_collect.py 中的 train() 即可, 生成模型为 model.pth 与 ark_material.onnx

依赖安装:

# 物品识别可以直接使用 opencv 内置的 dnn 模块
# 数量识别需要 PaddleOCR 或其它具有 std 功能的 ocr 引擎
pip install opencv-python, numpy, ppocr-onnx

需配合 ark_material.onnx 及 index_itemid_relation.json 使用.

具体使用方式见 demo.py