For CopyCat dataset from Kshitish's disk - Python script to convert .bytes to png

byteToImage.py

# Python script to convert .bytes to png

import io
from PIL import Image
import numpy as np
import sys
from array import array

def readimage(path):
    count = os.stat(path).st_size / 2
    with open(path, "rb") as f:
        return bytearray(f.read())

path = '/o/projects/copycat/kinect-conversion-test092119'+'/Alligator_behind_black_wall_color_257.bytes'
savepath = '/o/projects/copycat/kinect-conversion-test092119'+'/Alligator_behind_black_wall_color_257.png'
byteFile = readimage(path)
print(len(byteFile))
new_image = Image.frombytes("RGBA", (1920, 1080), bytes(byteFile))
new_image = new_image.convert("RGBA")
data = np.array(new_image)
red, green, blue, alpha = data.T
data = np.array([blue, green, red, alpha])
data = data.transpose()
new_image = Image.fromarray(data)

# image = Image.open(io.BytesIO(byteFile))                                                                                     
new_image.save(savepath)