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からダウンロードします．

• ResNet18
• ResNet34
• ResNet50

./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.

1. 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.