IDaRS

class IDaRS(backbone, num_classes=1)[source]

Retrieve the model and add custom preprocessing used in IDaRS paper.

Parameters

  • backbone (str) – Model name.

  • num_classes (int) – Number of classes output by model.

Methods

preproc

Define preprocessing steps.

Attributes

static preproc(img)[source]

Define preprocessing steps.

Parameters

img (np.ndarray) – An image of shape HWC.

Returns

An image of shape HWC.

Return type

img (torch.Tensor)