visualization

a collection of visualization operation for easier usage, check usage for a quick start.

New Features

2021/09/29

• Add pip installation
• Build a cleaner repo

Contents

Visualization Function

• Grid Attention Visualization
• Region Attention Visualization
• Draw Line Chart

Learning Notes Sharing

• Image Reading Difference

Installation

pip install visualize==0.5.0

Usage

Run Example

You can try example.py by cloning this repo for a quick start.

git clone https://github.com/rentainhe/visualization.git
python example.py

results will be saved to ./test_grid_attention and ./test_region_attention

Region Attention Visualization

download the example.jpg to any folder you like

$ wget https://github.com/rentainhe/visualization/blob/master/visualize/test_data/example.jpg

build the following python script for a quick start

import numpy as np
from visualize import visualize_region_attention

img_path="path/to/example.jpg"
save_path="example"
attention_retio=1.0
boxes = np.array([[14, 25, 100, 200], [56, 75, 245, 300]], dtype='int')
boxes_attention = [0.36, 0.64]
visualize_region_attention(img_path,
                           save_path=save_path, 
                           boxes=boxes, 
                           box_attentions=boxes_attention, 
                           attention_ratio=attention_retio,
                           save_image=True,
                           save_origin_image=True,
                           quality=100)

• img_path: where to load the original image
• boxes: a list of coordinates for the bounding boxes
• box_attentions: a list of attention scores for each bounding box
• attention_ratio: a special param, if you set the attention_ratio larger, it will make the attention map look more shallow. Just try!
• save_image=True: save the image with attention map or not, e.g., default: True.
• save_original_image=True: save the original image at the same time, e.g., default: True

Note that you can check Region Attention Visualization here for more details

Grid Attention Visualization

download the example.jpg to any folder you like

$ wget https://github.com/rentainhe/visualization/blob/master/visualize/test_data/example.jpg

build the following python script for a quick start

from visualize import visualize_grid_attention_v2
import numpy as np

img_path="./example.jpg"
save_path="test"
attention_mask = np.random.randn(14, 14)
visualize_grid_attention_v2(img_path,
                           save_path=save_path,
                           attention_mask=attention_mask,
                           save_image=True,
                           save_original_image=True,
                           quality=100)

• img_path: where the image you want to put an attention mask on.
• save_path: where to save the image.
• attention_mask: the attention mask with format numpy.ndarray, its shape is (H, W)
• save_image=True: save the image with attention map or not, e.g., default: True.
• save_original_image=True: save the original image at the same time, e.g., default: True

Note that you can check Attention Map Visualization here for more details