import numpy as np

class Normalize(grain.MapTransform):
    def map(self, element):
        element['image'] = element['image'].astype(np.float32) / 255.0
        return element

        __
        __