rise module

datumaro.components.algorithms.rise.flatmatvec(mat)

datumaro.components.algorithms.rise.expand(array, axis=None)

class datumaro.components.algorithms.rise.RISE(model, max_samples=None, mask_width=7, mask_height=7, prob=0.5, iou_thresh=0.9, nms_thresh=0.0, det_conf_thresh=0.0, batch_size=1)

Bases: object

Implements RISE: Randomized Input Sampling for Explanation of Black-box Models algorithm See explanations at: https://arxiv.org/pdf/1806.07421.pdf

__init__(model, max_samples=None, mask_width=7, mask_height=7, prob=0.5, iou_thresh=0.9, nms_thresh=0.0, det_conf_thresh=0.0, batch_size=1)

static split_outputs(annotations)

normalize_hmaps(heatmaps, counts)

apply(image, progressive=False)