| 262 | |
| 263 | |
| 264 | class Mask2Tensor(torch.nn.Module): |
| 265 | def __call__(self, x): |
| 266 | return torch.from_numpy(np.array(x)).long() |
| 267 | |
| 268 | class EncodeBitMap(torch.nn.Module): |
| 269 | def __init__(self, n=8, fill_value=0.5, ignore_label=0): |
no outgoing calls
no test coverage detected