ML-Danbooru: Anime image tags detector

修改

增加web-demo.py 参考 7eu7d7/ML-Danbooru-webui：ML-Danbooru 的 webui 扩展 (github.com) 实现gradio界面， demo_gradio_client_api.py

补充requirements.txt依赖

Introduction

An anime image tag detector based on modified ML-Decoder. Model trained with cleaned danbooru2021.

• Designed a new TResNet-D structure as backbone to enhance the learning of low-level features.
• Replace the ReLU in backbone with FReLU.
• Using learnable queries for transformer decoder.

Model Structure

Model-Zoo

https://huggingface.co/7eu7d7/ML-Danbooru https://huggingface.co/7eu7d7/ML-Danbooru/tree/main

Usage

Download the model and run below command:

python demo.py --data <path to image or directory> --model_name tresnet_d --num_of_groups 32 --ckpt <path to ckpt> --thr 0.7 --image_size 640 

Keep the image ratio invariant:

python demo.py --data <path to image or directory> --model_name tresnet_d --num_of_groups 32 --ckpt <path to ckpt> --thr 0.7 --image_size 640 --keep_ratio True

ML_CAFormer

python demo_ca.py --data <path to image or directory> --model_name caformer_m36 --ckpt <path to ckpt> --thr 0.7 --image_size 448