python用opencv读取图像如何做格式转换
在python中用cv2读取图像文件或摄像头,都能获取一帧图像,如
import cv2
frame1 = cv2.imread(“a.jpg”)
cap = cv2.VideoCapture(0)
rval, frame2 = cap.read()
上面获得的frame1和frame2都是numpy array格式的。但是有时候我们想要把它转成自己的格式,比如有时候可能需要调用C++函数,比如在darknet中,这时就要进行格式转换。这个方法有如下几种:
- 存成文件,再用相应的读文件方法读取
这种方法这里不谈,毕竟频繁地读写文件的方法在很多场合下是不可取的,其实速度还可以,但仍然不是一种完美的方式。 - 直接在python中做格式转换
既然知道frame是numpy array格式,当然可以直接做转换,但是在python中进行这种操作非常耗时。笔者测试了一下,转换一张416x416x3的图像大约需要3秒钟。这个速度甚至不如用文件转存的方式。 本文重点要说的,是通过C函数进行转换
首先,从numpy array获得C数据指针:
h,w,c=frame1.shape
if not frame1.flags[‘C_CONTIGUOUS’]:
frame1= np.ascontiguousarray(frame1, dtyp