def jpg2nv21(jpg):
yuv = cv2.cvtColor(jpg,cv2.COLOR_BGR2YUV_I420).flatten()
y_end =int(len(yuv)/3*2)
uv = yuv[y_end:]
uv = uv.reshape((2,-1))
uv =uv.transpose((1,0))
uv =uv.flatten()
yuv[y_end:] =uv
return yuv
jpg = cv2.imread(imgpath)
yuv = jpg2nv21(jpg)
yuv.tofile(save_path)