YUYV422 to PNG using OpenCV

yuv2png.py

#!/usr/bin/python3
from __future__ import print_function
import cv2
import numpy as np
import sys
import os.path

def c255(x):
    return min(max(int(x),0),255)

fd = open(sys.argv[1], 'rb')
img_str = fd.read()
fd.close()

img_array = np.asarray(bytearray(img_str), dtype=np.uint8)

height = 1080
width  = 1920

im = np.zeros((height,width,3), np.uint8)

c = 0
for j in range(0,height-1):
    for i in range(0,width-1,2):
        ind = c * 4
        c = c + 1
        
        p0 = img_array[ind+0] 
        p1 = img_array[ind+1] 
        p2 = img_array[ind+2] 
        p3 = img_array[ind+3] 
        
        y1 = 255/219.0*(p0 - 16)
        u  = 127/112.0*(p1 - 128)
        y2 = 255/219.0*(p2 - 16)
        v  = 127/112.0*(p3 - 128)
        
        r1 = c255(y1 + 1.402 * v)
        g1 = c255(y1 - 0.344 * u - 0.714*v)
        b1 = c255(y1 + 1.772 * u)

        r2 = c255(y2 + 1.402 * v)
        g2 = c255(y2 - 0.344 * u - 0.714*v)
        b2 = c255(y2 + 1.772 * u)
        
        im[j,i,0] = b1
        im[j,i,1] = g1
        im[j,i,2] = r1
        im[j,i+1,0] = b2
        im[j,i+1,1] = g2
        im[j,i+1,2] = r2

outname,ext = os.path.splitext(sys.argv[1])

cv2.imwrite(outname+'.png', im)