Palworld Terminal Pal Classifier (ONNX)

幻兽帕鲁终端界面「帕鲁图标」分类模型(MobileNetV3-Small),用于识别终端截图里每一格的帕鲁种类,配合网格切图即可统计数量。

本模型、训练流程与配套网页均由 AI 生成/编写;帕鲁素材(图标、名称、数据)版权归 Pocketpair, Inc. 所有。

文件

文件 说明
pal_classifier.onnx FP32 ONNX,输入 input [N,3,96,96],输出 logits [N,300]
pal_classifier_fp16.onnx FP16(内部半精度,I/O 仍是 float32),WebGPU 优先使用
labels.json 类别顺序:299 个内部 ID + __empty__(索引 299)
zh_names.json 内部 ID → 简体中文名(来自游戏本地化表)
train_classifier.py 训练脚本(合成数据 + MobileNetV3 + BN 校准 + ONNX 导出)
example_inference.py 最小推理示例(onnxruntime)

输入预处理

  • 裁剪出每个图标格子(约正方形,图标占 80–95%),缩放到 96×96。
  • RGB,归一化:x = (pixel/255 - 0.5) / 0.5。
  • 输出 300 维 logits,softmax 后取 argmax;__empty__ 表示空格。
  • 建议置信度阈值 0.35;低于阈值视为未识别。

训练与指标

  • 数据:从本地游戏 pak 提取的 299 个官方图标(Pal/Content/Pal/Texture/PalIcon/Normal/T_<id>_icon_normal), 程序合成圆形半透明底板、场景背景、缩放/旋转/亮度/模糊/JPEG 压缩、选中框、锁定角标、空槽等。
  • 模型:MobileNetV3-Small(ImageNet 预训练),96×96,300 类。
  • 训练:AdamW + cosine,6 epochs,AMP;训练后对 BatchNorm 做累积校准 (torchvision 的 BN 动量 0.01 会让 running stats 滞后,直接 eval 会退化)。
  • 指标:合成验证集 top-1 **99.8%**;真实终端截图 30/30 格识别正确(含空格判断)。

用法

import json, numpy as np, onnxruntime as ort
from PIL import Image

sess = ort.InferenceSession("pal_classifier.onnx", providers=["CPUExecutionProvider"])
labels = json.load(open("labels.json"))["labels"]
zh = json.load(open("zh_names.json"))

img = Image.open("cell.png").convert("RGB").resize((96, 96), Image.LANCZOS)
x = np.asarray(img, np.float32) / 255.0
x = ((x - 0.5) / 0.5).transpose(2, 0, 1)[None]

logits = sess.run(None, {"input": x})[0][0]
idx = int(logits.argmax())
print(labels[idx], zh.get(labels[idx], ""))

浏览器端:onnxruntime-web,WebGPU 优先、WebGL / WASM 回退;参考 https://lingxingmiao.github.io/pal-terminal-scanner/(识别/打标)与 https://lingxingmiao.github.io/pal-terminal-scanner/breeding.html(自动配种)。

训练脚本说明

train_classifier.py 默认从 C:\Users\FengMang\Desktop\pal_work\icons_png 读取 299 张图标 PNG (每张 128×128 RGBA,文件名 = 内部 ID)。游戏图标不随本仓库分发,请自行从自己的游戏安装中提取; 路径可用环境变量或直接修改 ICON_DIR。

版权与免责声明

  • Palworld 及其帕鲁图标、名称、配种数据等素材 **© Pocketpair, Inc.**,本仓库不包含游戏原始资源。
  • 本模型仅用于粉丝研究、学习与个人工具,禁止商用;与 Pocketpair 无关联、未获其背书。
  • 模型/脚本/网页由 AI 生成,可能存在错误,使用风险自负。

许可证

  • 本模型的权重与示例代码:CC BY-NC 4.0 (Creative Commons Attribution-NonCommercial 4.0 International,署名—非商业性使用 4.0 国际)。
  • Palworld 及其帕鲁图标、名称、配种数据等素材 **© Pocketpair, Inc.**, 不属于上述许可证的授权范围,仅作粉丝研究用途。

引用

如果使用,请注明: Palworld Terminal Pal Classifier (AI-generated ONNX model), fan project, Palworld assets © Pocketpair, Inc., licensed CC BY-NC 4.0

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support