Skip to content

Latest commit

 

History

History
24 lines (19 loc) · 1.19 KB

README.md

File metadata and controls

24 lines (19 loc) · 1.19 KB

D-RISE implementation

Implementation is based on D-RISE paper. Random mask resolution is (16, 16). Number of masks is 10000. Single target bbox is picked per image, if confident prediction is available.

Original work generates masks to sample input image. Current approach also generates random masks to sample from the feature map space. Masks are generated for middle FPN scale (target_scale_id = len(backbone_out) // 2) and then rescaled to other upstream and downstream scale levels. This might be not the best approach, but it produces decent baseline. See d_rise for more details.

Performance of the implementation is far from being optimal, but enough to perform initial investigation.

Please note that D-RISE might fail in some cases to generate high quality saliency map, due to bad luck of random mask generation.

Installation

For installation follow instructions in mmdetection get_started. To reproduce the results, use versions specified in requirements.txt (optional).

Usage

python src/run.py path_to_configs path_to_checkpoints path_to_data path_to_output