pass the code to the another device

[Agito555]

class gamma_transform(object):
def init(self,gamma_range):
self.gamma_range=gamma_range
def call(self,sample):
image, mask = sample
gamma=np.random.uniform(low=self.gamma_range[0],high=self.gamma_range[1])
image=np.array(image)
image=image/255
image=image**gamma
image=image*255
image[image>255]=255
image[image<0]=0
image=image.astype(np.uint8)
image=Image.fromarray(image)
return image, mask