(self, n=8, fill_value=0.5, ignore_label=0)
| 267 | |
| 268 | class EncodeBitMap(torch.nn.Module): |
| 269 | def __init__(self, n=8, fill_value=0.5, ignore_label=0): |
| 270 | super().__init__() |
| 271 | self.n = n |
| 272 | self.fill_value = fill_value |
| 273 | self.ignore_label = ignore_label |
| 274 | |
| 275 | def __call__(self, x): |
| 276 | ignore_mask = x == self.ignore_label |
nothing calls this directly
no outgoing calls
no test coverage detected