(
self,
level=1,
num_classes=None,
adjacency_mode='radius',
k_max=30,
radius=1,
use_batch=True,
centroid_mode='iou',
centroid_level=1,
smooth_affinity=True)
| 118 | _CENTROID_MODES = ['iou', 'ratio-product'] |
| 119 | |
| 120 | def __init__( |
| 121 | self, |
| 122 | level=1, |
| 123 | num_classes=None, |
| 124 | adjacency_mode='radius', |
| 125 | k_max=30, |
| 126 | radius=1, |
| 127 | use_batch=True, |
| 128 | centroid_mode='iou', |
| 129 | centroid_level=1, |
| 130 | smooth_affinity=True): |
| 131 | assert adjacency_mode.lower() in self._ADJACENCY_MODES, \ |
| 132 | f"Expected 'mode' to be one of {self._ADJACENCY_MODES}" |
| 133 | assert centroid_mode.lower() in self._CENTROID_MODES, \ |
| 134 | f"Expected 'mode' to be one of {self._CENTROID_MODES}" |
| 135 | self.level = level |
| 136 | self.num_classes = num_classes |
| 137 | self.adjacency_mode = adjacency_mode.lower() |
| 138 | self.k_max = k_max |
| 139 | self.radius = radius |
| 140 | self.use_batch = use_batch |
| 141 | self.centroid_mode = centroid_mode.lower() |
| 142 | self.centroid_level = centroid_level |
| 143 | self.smooth_affinity = smooth_affinity |
| 144 | |
| 145 | def _process(self, nag): |
| 146 | # Skip the transform. This mechanism can be useful for skipping |
nothing calls this directly
no outgoing calls
no test coverage detected