MeanIoU

class MeanIoU

Metric that calculates the mean IoU metric for each class.

See more: https://en.wikipedia.org/wiki/Jaccard_index

Parameters

threshold: float, default: 0.5

prediction value per pixel above which the pixel is considered True.

smooth: float, default: 1e-3

smoothing factor to prevent division by zero when the mask is empty.

average: str, default: none

The method for averaging over the classes. If none, returns the result per class.

__init__(*args, threshold: float = 0.5, smooth=0.001, average: str = 'none', **kwargs)

__new__(*args, **kwargs)

Attributes

MeanIoU.required_output_keys

Methods

MeanIoU.attach(engine, name[, usage])	Attaches current metric to provided engine.
MeanIoU.completed(engine, name)	Helper method to compute metric's value and put into the engine.
MeanIoU.compute()	Compute metric value.
MeanIoU.detach(engine[, usage])	Detaches current metric from the engine and no metric's computation is done during the run.
MeanIoU.is_attached(engine[, usage])	Checks if current metric is attached to provided engine.
MeanIoU.iteration_completed(engine)	Helper method to update metric's computation.
MeanIoU.reset()	Reset metric state.
MeanIoU.started(engine)	Helper method to start data gathering for metric's computation.
MeanIoU.update(output)	Update metric with batch of samples.

previous

MeanDice.update

next

MeanIoU.required_output_keys