anime image classification
Project description
DeepDanbooru
动漫图片分类模型
安装
pip install deepdanbooru-onnx
使用
from deepdanbooru_onnx import DeepDanbooru, process_image
from PIL import Image
import numpy as np
danbooru = DeepDanbooru()
print(danbooru('test.jpg'))
img = Image.open('test.jpg')
print(danbooru(img))
img = process_image(img) # iamge to ndarray
print(danbooru(img))
print(list(danbooru(['test1.jpg','test2.jpg'])))
TODO
- 提供轻量化模型
- 结果缓存
- 提供WEB接口
- 模型量化
- 精度更高的模型(slow)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
deepdanbooru_onnx-0.0.7.tar.gz
(16.5 kB
view hashes)
Built Distribution
Close
Hashes for deepdanbooru_onnx-0.0.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5b16dd98f124c60a5a8aef75dca65c4362d6c79e0d000589187f9e38b4e822f5 |
|
MD5 | 040f5e6a44e4657b1c8ffa42d25e74dd |
|
BLAKE2b-256 | 99a62f5d072923fa90e050b5af80e576ecee3682055eb1e61168cfac19079374 |