def __init__(self, root, split="train", is_transform=False, is_augment=False): self.root = root self.split = split self.img_size = (360, 480) # (h, w) self.is_transform = is_transform self.mean = np.array([104.00699, 116.66877, 122.67892]) self.n_classes = 12 self.files = collections.defaultdict(list) self.joint_augment_transform = None self.is_augment = is_augment if self.is_augment: self.joint_augment_transform = Compose([ # RandomSized(int(480)), RandomRotate(degree=10), RandomHorizontallyFlip(), ]) # file_list = os.listdir(root + '/' + split) file_list = glob.glob(root + '/' + split + '/*.png') file_list.sort() self.files[split] = file_list
def __init__(self, root, split="train", is_transform=False, is_augment=False): self.root = root self.split = split self.img_size = (360, 480) # (h, w) self.is_transform = is_transform self.mean = np.array([104.00699, 116.66877, 122.67892]) self.n_classes = 2 self.files = collections.defaultdict(list) self.joint_augment_transform = None self.is_augment = is_augment if self.is_augment: self.joint_augment_transform = Compose([ # RandomSized(int(min(self.img_size)/0.875)), # RandomCrop(self.img_size), RandomRotate(degree=10), RandomHorizontallyFlip(), ]) # file_list = os.listdir(root + '/' + split) file_list = glob.glob(root + '/' + split + '/*/*.png') # print('file_list:', file_list) file_list.sort() self.files[split] = file_list if self.split == 'train': self.input_shape = (64, 64) # self.input_shape = (128, 256) elif self.split == 'val': self.input_shape = (128, 256)
def __init__(self, root, split="train", is_transform=False, is_augment=False): self.root = root self.split = split self.img_size = [360, 480] self.is_transform = is_transform self.mean = np.array([104.00699, 116.66877, 122.67892]) self.n_classes = 13 self.files = collections.defaultdict(list) self.joint_augment_transform = None self.is_augment = is_augment if self.is_augment: self.joint_augment_transform = Compose([ RandomHorizontallyFlip(), RandomRotate(degree=90) # transforms.ToTensor() ]) file_list = os.listdir(root + '/' + split) self.files[split] = file_list