Skip to main content

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)

Uploaded source

Built Distribution

deepdanbooru_onnx-0.0.7-py3-none-any.whl (16.8 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page