必要なモデルファイル、平均画像ファイル、タグ一覧はnico-opendata.jpからダウンロードしてください。
依存ライブラリのインストール
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
MIT License (see LICENSE
file).