hayamdk/tag_prediction_example

nico-illust tag prediction

必要なモデルファイル、平均画像ファイル、タグ一覧はnico-opendata.jpからダウンロードしてください。

USAGE

依存ライブラリのインストール

pip install -r requirements.txt

CPUで実行

python predict_tag.py \
  --gpu=-1 \
  nico_illust_tag_v0.dump \
  mean.dump \
  character_series.txt \
  http://lohas.nicoseiga.jp/thumb/605863i
# tag: 東方 / score: 1.0
# と出力

GPUで実行

python predict_tag.py \
  --gpu=0 \
  nico_illust_tag_v0.dump\
  mean.dump \
  character_series.txt \
  http://lohas.nicoseiga.jp/thumb/605863i

License

MIT License (see LICENSE file).