| 4 | from torchvision.transforms.functional import to_tensor, normalize |
| 5 | |
| 6 | class FieldDef(object): |
| 7 | def __init__(self, field, index, bitshift, bitmask, labels): |
| 8 | self.field = field |
| 9 | self.index = index |
| 10 | self.bitshift = bitshift |
| 11 | self.bitmask = bitmask |
| 12 | self.labels = labels |
| 13 | |
| 14 | class MultiSegmentDataset(object): |
| 15 | ''' |