mirror of
https://github.com/gunchev/home_bin.git
synced 2025-11-20 12:15:46 +00:00
Doncho Gunchev
c45966f597
From https://shrex999.wordpress.com/2013/07/31/yuv-to-rgb-python-imaging-library/
58 lines
1.6 KiB
Python
Executable file
58 lines
1.6 KiB
Python
Executable file
# Author : sherin.s@gmail.com (Sherin Sasidhan)
|
|
# Date : 05-Oct-2010
|
|
|
|
import Image
|
|
import sys
|
|
from struct import *
|
|
import array
|
|
|
|
if len(sys.argv) != 4:
|
|
print "***** Usage syntax Error!!!! *****\n"
|
|
print "Usage:"
|
|
print "python <script> <.yuv file uyvy422> <width>> <height> "
|
|
sys.exit(1) # exit
|
|
else:
|
|
pass
|
|
|
|
image_name = sys.argv[1]
|
|
width = int(sys.argv[2])
|
|
height = int(sys.argv[3])
|
|
|
|
y = array.array('B')
|
|
u = array.array('B')
|
|
v = array.array('B')
|
|
|
|
f_uyvy = open(image_name, "rb")
|
|
f_uv = open(image_name, "rb")
|
|
f_uv.seek(width*height, 1)
|
|
|
|
image_out = Image.new("RGB", (width, height))
|
|
pix = image_out.load()
|
|
|
|
print "width=", width, "height=", height
|
|
|
|
for i in range(0,height):
|
|
for j in range(0, width/2):
|
|
u = ord(f_uyvy.read(1));
|
|
y1 = ord(f_uyvy.read(1));
|
|
v = ord(f_uyvy.read(1));
|
|
y2 = ord(f_uyvy.read(1));
|
|
|
|
B = 1.164 * (y1-16) + 2.018 * (u - 128)
|
|
G = 1.164 * (y1-16) - 0.813 * (v - 128) - 0.391 * (u - 128)
|
|
R = 1.164 * (y1-16) + 1.596*(v - 128)
|
|
pix[j*2, i] = int(R), int(G), int(B)
|
|
|
|
B = 1.164 * (y2-16) + 2.018 * (u - 128)
|
|
G = 1.164 * (y2-16) - 0.813 * (v - 128) - 0.391 * (u - 128)
|
|
R = 1.164 * (y2-16) + 1.596*(v - 128)
|
|
pix[j*2+1, i] = int(R), int(G), int(B)
|
|
|
|
######################################################
|
|
# B = 1.164(Y - 16) + 2.018(U - 128)
|
|
# G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
|
|
# R = 1.164(Y - 16) + 1.596(V - 128)
|
|
######################################################
|
|
|
|
#image_out.save("out.bmp")
|
|
image_out.show()
|