def __init__(self, dataset, config):
super().__init__(dataset, config)
self.label_dic = dataset.label_dic
self.joint_transform_list = [
None,
imutils.RandomHorizontalFlip(),
imutils.RandomResizeLong(512, 768),
imutils.RandomCrop(448),
None,
]
def __init__(self, dataset, config):
self.image_ids = np.copy(dataset.image_ids)
self.dataset = dataset
self.config = config
self.mean = (0.485, 0.456, 0.406)
self.std = (0.229, 0.224, 0.225)
self.joint_transform_list = [
None,
imutils.RandomHorizontalFlip(),
imutils.RandomResizeLong(512, 832),
imutils.RandomCrop(448),
None,
]
self.img_transform_list = [
transforms.ColorJitter(brightness=0.3,
contrast=0.3,
saturation=0.3,
hue=0.1), np.asarray, None,
imutils.Normalize(mean=self.mean, std=self.std), imutils.HWC_to_CHW
]