def __init__(self, renderer, image_paths, num_occlusions, split=pr.TRAIN):
super(DomainRandomizationProcessor, self).__init__()
self.copy = pr.Copy()
self.render = pr.Render(renderer)
self.augment = RandomizeRenderedImage(image_paths, num_occlusions)
preprocessors = [pr.ConvertColorSpace(pr.RGB2BGR), pr.NormalizeImage()]
self.preprocess = SequentialProcessor(preprocessors)
self.split = split
def __init__(self):
super(PipelineWithThreeChannelsPlus, self).__init__()
self.add(lambda a, b: (a, b))
self.add(pr.ControlMap(pr.Copy(), [0], [2], keep={0: 0}))
self.add(pr.ControlMap(SumTwoValues(), [0, 1], [0]))
def __init__(self):
super(PipelineWithTwoChannels, self).__init__()
self.add(lambda x: x)
self.add(pr.ControlMap(pr.Copy(), [0], [1], keep={0: 0}))