kosuke1701/illust-search

イラストの検索システムを作るプロジェクト。

Getting started

顔ベクトル計算前処理

• AnimeCVのインストール

  pip install git+https://github.com/kosuke1701/AnimeCV.git
  

• イラストからの顔検知およびembedding計算

  python encode_face_vectors.py --target-fn <TARGET> --image-root <ROOT_DIR> --cuda
  

  • <TARGET>は1列目がイラストIDで、2列目が<ROOT_DIR>からのイラスト画像の相対パスであるようなTSVファイル。
  • embeddingを保存したデータベースファイルvectors.sqlが作成される。
    • 計算済みのデータベースは以下からダウンロードできる
    • https://github.com/kosuke1701/illust-search/releases/download/0.0/vectors.sql
  • データベース中にイラストIDが既存の画像は処理しないので、<TARGET>が更新された場合は単純に上記コマンドを再度実行すればよい。

ScaNNによる検索のための索引作成

• 以下のNotebookを実行してScaNNの索引を作成し保存する。

  • 計算済みの索引は以下からダウンロードできる
  • https://github.com/kosuke1701/illust-search/releases/download/0.0/scann_save_dir.zip

  Notebookを開く

pixiv顔検索

Notebookを開く