OV0142双摄主板（续）

代码整合实时读取

废话不多说直接上代码！详情跟请看上一篇

import usb.core
import usb.util
import sys
import threading
import time
import cv2
from PIL import Image
from numpy import *
 
VID=0x05a9
PID=0x8065

def xianfu(x):
    if(x>=0 and x<=255):
        return x
    elif(x<0):
        return 0
    else:
        return 255  

def readYuvList(data, width, height):
    C=0
    Y1 = zeros((height, width//2), uint8, 'C')
    Y2 = zeros((height, width//2), uint8, 'C')
    U = zeros((height, width//2), uint8, 'C')
    V = zeros((height, width//2), uint8, 'C')
    C+=12
    RGBarray=zeros((height, width,3), uint8, 'C')
    for m in range(height):
        for n in range(width//2):
            # U[m, n] = ord(fp.read(1))
            # Y1[m, n] = ord(fp.read(1))
            # V[m, n] = ord(fp.read(1))
            # Y2[m, n] = ord(fp.read(1))
            y1 = data[C]
            C+=1
            u = data[C]
            C+=1
            y2 = data[C]
            C+=1
            v = data[C]
            C+=1
            if(C%2048==0):
                C+=12
            r = (y1 + (u - 128) + ((104*(u - 128))>>8))
            g = (y1 - (89*(v - 128)>>8) - ((183*(u - 128))>>8))
            b = (y1 + (v - 128) + ((199*(v - 128))>>8))

            RGBarray[m,n*2,0]=xianfu(r)   
            RGBarray[m,n*2,1]=xianfu(g)   
            RGBarray[m,n*2,2]=xianfu(b)   

            r = (y2 + (u - 128) + ((104*(u - 128))>>8))
            g = (y2 - (89*(v - 128)>>8) - ((183*(u - 128))>>8))
            b = (y2 + (v - 128) + ((199*(v - 128))>>8))

            RGBarray[m,n*2+1,0]=xianfu(r)   
            RGBarray[m,n*2+1,1]=xianfu(g)   
            RGBarray[m,n*2+1,2]=xianfu(b)              
    return RGBarray

all_devs=usb.core.find(find_all=True)

for dev in all_devs:
    print(dev)

cfg = dev.get_active_configuration()
intf = cfg[(0,0)]
 
print(intf)
 
write_ep = usb.util.find_descriptor(
    intf,
    custom_match = \
    lambda e: \
        usb.util.endpoint_direction(e.bEndpointAddress) == \
        usb.util.ENDPOINT_OUT
)
 
read_ep = usb.util.find_descriptor(
    intf,
    custom_match = \
    lambda e: \
        usb.util.endpoint_direction(e.bEndpointAddress) == \
        usb.util.ENDPOINT_IN
    )
 
def write_endpoint_thread():
    while True:
        try:
            write_ep.write("123")
        except:
            print("write error")
    
 
def read_endpoint_thread():
    while True:
        data = read_ep.read(643780)
        size = len(data)
        starttime = time.clock()
        if size > 30000:
            img=readYuvList(data.tolist(),800,400)
            cv2.imshow('0',img)
            endtime = time.clock()
            print(endtime - starttime)
            starttime=endtime
            k=cv2.waitKey(1)
            if k == 27:
                #通过esc键退出摄像
                cv2.destroyAllWindows()
                break
    cv2.destroyWindow('0')
 
    
 
def main():
    r = threading.Thread(target=read_endpoint_thread)
    w = threading.Thread(target=write_endpoint_thread)
    r.start()
    #w.start()
    r.join()
    #w.join()
    dev.reset()
 
if __name__ == '__main__':
    main()

这里加了opencv实时显示图片（numpy完美衔接）

运行这个代码前我就一直怀疑它的帧率能不能配的上视频这个名词。果然没令我失望，2s一帧。python的运行效率懂的都懂。但是python也有强大的解决方案，运用Cpython将其打包一下底层的一些加减乘除的运算都用C的模式计算大大提高了效率。大佬传送门 看看这个教程跟着做就好了，方便快捷。在查找的时候还看到一个pypy说是效率更高。后面再试试。

目前使用Cpython编译后打到0.3s一帧，虽然还是差了很多（Demo上位机1s30帧），算法有待优化。