Author: Masahiro Mitsuhara
Maintainer: Tsubasa Hirakawa
Anaconda (3-5.3.1) を用いてPython環境を構築しています.
- torch (0.4.1)
- torchvision (0.2.1)
- opencv (3.4.2)
- PyQt5 (5.9.2)
CUB-200-2010 datasetを下記のwebページからダウンロードし,保存します. CBU-200-2010 dataset
学習済みモデルを下記のURLからダウンロードします.
./atteditor/config.py内のパスを各自の環境に合わせて書き換えます.
# 本プロジェクトのディレクトリまでのパス
PROJECT_ROOT = "/path/to/AttentionEditorABN"
# データセット(画像)を保存しているディレクトリまでのパス
IMAGE_DIR = "/path/to/CUB-200-2010_dataset"
# アプリケーションに表示するネットワーク名
MODEL_NAME = "ABN ResNet 50"
# 学習済みモデルのパス
CHECKPOINT_FILE = "/path/to/pretrained_model_file"
下記のコマンドを実行することでアプリケーションが起動します.
python3 main.py
左側のウィンドウは使用する画像データを指定するためのウィンドウです.
右側のウィンドウは指定したネットワークモデルを使用して推論・推論結果を表示するためのウィンドウです.
**の部分で画像・アテンションマップの表示,修正を行います.
If you find useful for this application, please cite the following reference.
- M. Mitsuhara, H. Fukui, Y. Sakashita, T. Ogata, T. Hirakawa, T. Yamashita, H. Fujiyoshi, "Embedding Human Knowledge into Deep Neural Network via Attention Map," arXiv preprint, arXiv:1905.03540, 2019.