模型

private-detector.onnx

一个用于检测淫秽图像的图像分类器：https://github.com/bumble-tech/private-detector：

1. 从此处下载保存的模型：https://storage.googleapis.com/private_detector/private_detector_with_frozen.zip
2. 安装 tensorflow-onnx
3. 运行 python -m tf2onnx.convert --saved-model model/saved_model --output private-detector.onnx

toxic-comment

该模型是 DistilBERT 模型的微调版本，用于分类有害评论：https://huggingface.co/martin-ha/toxic-comment-model

1. 运行 optimum-cli export onnx --model martin-ha/toxic-comment-model toxic-comment/