fufufukakaka's picture
Update README.md
899c59d
metadata
license: mit
tags:
  - vision
  - image-classification
metrics:
  - accuracy
widget:
  - src: >-
      https://huggingface.co/datasets/fufufukakaka/autotrain-data-pokemon-image-classification-5/resolve/main/raw/image_folders/auto/imgs/%E3%83%8F%E3%83%83%E3%82%B5%E3%83%A0/0.png
    example_title: ハッサム
  - src: >-
      https://huggingface.co/datasets/fufufukakaka/autotrain-data-pokemon-image-classification-5/resolve/main/raw/image_folders/auto/imgs/%E3%83%86%E3%83%84%E3%83%8E%E3%83%84%E3%83%84%E3%83%9F/0.png
    example_title: テツノツツミ
  - src: >-
      https://huggingface.co/datasets/fufufukakaka/autotrain-data-pokemon-image-classification-5/resolve/main/raw/image_folders/auto/imgs/%E3%83%8C%E3%83%A1%E3%83%AB%E3%82%B4%E3%83%B3/3.png
    example_title: ヌメルゴン

ポケモンの選出画面での画像を入力として、そのポケモンを識別するモデルです。

microsoft/swin-base-patch4-window7-224-in22k を元にして fine-tuning しています。

Repo: https://github.com/fufufukakaka/poke_battle_logger