Skip to content
master
Go to file
Code

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

README.md

nico-illust tag prediction

必要なモデルファイル、タグ一覧は nico-opendata.jp からダウンロードしてください。 解凍して出来たディレクトリ下の v1/ 下にある model.npz 及び tags.txt をカレントディレクトリにコピーして使います。

USAGE

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

pip install -r requirements.txt

CPUで実行

python predict_tag.py \
  --gpu=-1 \
  --tags=tags.txt
  --model=model.npz
  http://lohas.nicoseiga.jp/thumb/4313120i
# tag: 川内改二 / score: 0.9832866787910461
# tag: 艦これ / score: 0.9811543226242065
# tag: 夜戦忍者 / score: 0.934027910232544
#    :
# と出力

GPUで実行

python predict_tag.py \
  --gpu=0 \
  --tags=tags.txt
  --model=model.npz
  http://lohas.nicoseiga.jp/thumb/4313120i

License

MIT License (see LICENSE file).

About

No description, website, or topics provided.

Resources

License

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.