Repository files navigation

This is the author implementation ofSHAP-Based Interpretable Object Detection Method for Satellite Imagery. The implementation of the object detection model (YOLOv3) is based onPytorch_YOLOv3. The framework of the proposed method can be applied to any differentiable object detection model.

Please see the paper for details on the results of the evaluation, regularization, and data selection methods.

• Python 3.6.3+
• Numpy
• OpenCV
• Matplotlib
• Pytorch 1.2+
• Cython
• Cuda (verified as operable: v10.2)
• Captum (verified as operable: v0.4.1)

optional:

• tensorboard
• tensorboardX
• CuDNN

download the pretrained file from the author's project page:

$ mkdir weights$cd weights/$ bash ../requirements/download_weights.sh

Please see the test.ipynb

Hiroki Kawauchi, Takashi Fuse

[Paper][Original Implementation]