--- tags: - image-classification - timm - transformers - animetimm - dghs-imgutils library_name: timm license: gpl-3.0 datasets: - deepghs/danbooru-wdtagger-v4a-w640-ws-fullxx-cls base_model: - animetimm/swinv2_base_window8_256.dbv4a-fullxx --- # Anime Classifier swinv2_base_window8_256.dbv4a-fullxx-cls ## Model Details - **Model Type:** Single-Label Image classification / feature backbone - **Model Stats:** - Params: 96.6M - FLOPs / MACs: 121.6G / 60.7G - Image size: train = 448 x 448, test = 448 x 448 - **Dataset:** [deepghs/danbooru-wdtagger-v4a-w640-ws-fullxx-cls](https://huggingface.co/datasets/deepghs/danbooru-wdtagger-v4a-w640-ws-fullxx-cls) - Labels Count: 9453 ## Results | # | Top-1 | Top-5 | Macro (F1/P/R) | Micro (F1/P/R) | |:----------:|:-------:|:-------:|:---------------------:|:---------------------:| | Validation | 90.49% | 95.44% | 0.881 / 0.901 / 0.885 | 0.905 / 0.905 / 0.905 | | Test | 90.36% | 95.34% | 0.880 / 0.901 / 0.884 | 0.904 / 0.904 / 0.904 | You can find label list in [selected_tags.csv](https://huggingface.co/animetimm/swinv2_base_window8_256.dbv4a-fullxx-cls/resolve/main/selected_tags.csv).