ochw - 手写识别库

ochw 是一个基于卷积神经网络的手写汉字识别模型，基于 CASIA-HWDB 1.0 数据集训练，可识别 3740 个汉字和小部分符号。

安装

cargo add ochw

模型模型

wget https://huggingface.co/rkingzhong/ochw/resolve/main/ochw_mobilenetv2_fp16.safetensors?download=true

使用

以 知 字图片为例子。

let weights = include_bytes!("ochw_mobilenetv2_fp16.safetensors");
let worker = Inference::load_model(weights)?;
let labels = worker.get_labels()?;
let image = vec![];//Vec<u8> data
let output = worker.predict(image,Some(10))?;

println!("{:?}",output);

结果如下:

1. Class 知: 96.17%
2. Class 勉: 0.62%
3. Class 处: 0.28%
4. Class 贮: 0.23%
5. Class 矩: 0.12%