Get class names of current dataset. Args: classes (Sequence[str] | str | None): If classes is None, use default C

We organize the masks as following: - mask1: RTL - mask2: RTL - mask3: RTL - mask4: RTL

Get one ground truth segmentation map for evaluation.

Get one ground truth segmentation map for evaluation.

Call function to collect indexes. Args: dataset (:obj:`MultiImageMixDataset`): The dataset. Returns: list: i

Call function to collect indexes. Args: dataset (:obj:`MultiImageMixDataset`): The dataset. Returns: list: i

Return metrics for SAGAN experiment on TPU, CPU, or GPU. When training on TPUs, this function should be executed on the CPU. Args: rea

Returns the update operators for the weights. Please run this operators in session or add this to training operation. [Return

Sample points for testing. Find ``num_points`` most uncertain points from ``uncertainty_map``. Args: seg_logits (Tensor)

Sample points for testing. Find ``num_points`` most uncertain points from ``uncertainty_map``. Args: seg_logits (Tensor)

Sample points for training. Sample points in [0, 1] x [0, 1] coordinate space based on their uncertainty. The uncertainties are calcu

Sample points for training. Sample points in [0, 1] x [0, 1] coordinate space based on their uncertainty. The uncertainties are calcu

Hue distortion.

Hue distortion.

RGB-color encoding to grayscale labels.

RGB-color encoding to grayscale labels.

Draw `result` over `img`. Args: img (str or Tensor): The image to be displayed. seg (Tensor): The semantic segmentation results t

Draw `result` over `img`. Args: img (str or Tensor): The image to be displayed. seg (Tensor): The semantic segmentation results t

Initialize random seed. If the seed is not set, the seed will be automatically randomized, and then broadcast to all processes to prevent som

Initialize weight of later layer.

Initialize weight of later layer.

Load annotation from directory. Args: img_dir (str): Path to image directory img_suffix (str): Suffix of images.

Load annotation from directory. Args: img_dir (str): Path to image directory img_suffix (str): Suffix of images.

Compute segmentation loss.

Compute segmentation loss.

Compute ``pam_cam``, ``pam``, ``cam`` loss.

Compute segmentation loss.

Compute segmentation loss.

Compute ``pam_cam``, ``pam``, ``cam`` loss.