in dataset/transform.py [0:0]
def __call__(self, data):
"""Perform transformation
Arguments:
data {dict} -- frame data
"""
if 'image' not in list(data.keys()):
return data
# get image from all data
img = data['image']
# channel last to channel first
img = np.transpose(img, [2, 0, 1])
# normalization
img -= self.mean
img /= self.std
data.update({'image': img})
return data