dataset.py 文件源码

def augment_data(self, v_img_):
        # TODO: create a separate class for augmentation
        imsize = self.imsize
        for img in v_img_:
            #randomly flip
            if randint(0,65535)>32767:
                img[:,:,:] = img[:,:,::-1]
            #adjust gamma
            gamma = np.random.uniform(0.77,1.3, size=(imsize[0],1,1))
            img **= gamma
            #randomly crop image
            coinflip = lambda:randint(0,1)
            if coinflip():
                idx = lambda x:slice(None,x)
            else:
                idx = lambda x:slice(-x,None)
            if coinflip():
                n = randint(0,imsize[1]/4)
                idx = (slice(None), idx(n), slice(None))
            else:
                n = randint(0,imsize[2]/4)
                idx = (slice(None), slice(None), idx(n))
            img[idx] = 0.
        return v_img_