Functionmap_get_pairsget map pairs Args: labels_4D : labels, shape [N, C, H, W] probs_4D : probabilities, shape [N, C, H, W] radius : the square radius Return:
loss/rmi_utils.py:15
Functionmap_get_pairs_regionget map pairs Args: labels_4D : labels, shape [N, C, H, W]. probs_4D : probabilities, shape [N, C, H, W]. radius : The side length of the squa
loss/rmi_utils.py:59