chinoll / deepdanbooru_onnx

Easy-to-use danbooru anime image classification model

EN

DeepDanbooru

动漫图片分类模型

安装

pip install deepdanbooru-onnx

使用

from deepdanbooru_onnx import DeepDanbooru, process_image
from PIL import Image
import numpy as np
danbooru = DeepDanbooru()
#usage 1
print(danbooru('test.jpg'))
img = Image.open('test.jpg')
print(danbooru(img))
#usage 2
img = process_image(img) # iamge to ndarray
print(danbooru(img))
#usage 3
print(list(danbooru(['test1.jpg','test2.jpg'])))

TODO

• 提供轻量化模型
• 结果缓存
• 提供WEB接口
• 模型量化
• 精度更高的模型(slow)