Hey, I have been searching for something like this extension, great work.
There is a bug when the image is in range 0..1 floats, it is not saved correctly.
There should probably be some checking: if img.dtype = np.float32 or np.float64 then img * 255.0
