说明文档

Camie 标注器 ONNX 导出版本

如何使用

pip install -U dghs-imgutils

from imgutils.tagging import get_camie_tags

rating, tags, chars = get_camie_tags('image.png')
print(rating)
print(tags)
print(chars)

更多详情请参阅我们的文档。

共导出 2 个 ONNX 模型。

名称	参数量	计算量	输入尺寸	特征维度	类别数	阈值	F1 (MI/MA)	精确率 (MI/MA)	召回率 (MI/MA)	创建时间
initial	214.1M	64.7G	512	1280	70527	0.2576	0.584 / 0.383	0.498 / 0.382	0.706 / 0.426	2025-03-11
refined	334.0M	68.1G	512	2560	70527	0.2616	0.589 / 0.387	0.507 / 0.389	0.702 / 0.427	2025-03-11

作者 deepghs

image-classification dghs-imgutils

↓ 0 ♥ 0

创建时间: 2025-03-28 19:02:48+00:00

更新时间: 2025-03-30 18:13:56+00:00

.gitattributes

README.md

initial/initial_emb_to_pred.onnx ONNX

initial/meta.json

initial/model.onnx ONNX

initial/model.safetensors

initial/model_initial_only.onnx ONNX

initial/preprocess.json

initial/refined_emb_to_pred.onnx ONNX

initial/selected_tags.csv

initial/threshold.json

models.parquet

refined/initial_emb_to_pred.onnx ONNX

refined/meta.json

refined/model.onnx ONNX

refined/model.safetensors

refined/model_initial_only.onnx ONNX

refined/preprocess.json

refined/refined_emb_to_pred.onnx ONNX

refined/selected_tags.csv

refined/threshold.json