正如您所注意到的,OpenCV以BGR格式读取图像,而QImage以RGB格式读取。在第一种方法中,您在不进行转换的情况下将其转换为QImage,然后使用
rgbSwapped()
方法进行转换。
通过测试第一种方法,我得到:
1000 loops, best of 5: 291 usec per loop
在第二种方法中,您尝试在将其转换为QImage之前进行操作,但当我执行它时,我得到以下错误,假设您也会遇到相同的错误。
Traceback (most recent call last):
File "xxxx.py", line 18, in <module>
qImg = QtGui.QImage(src.data, w, h, bytesPerLine, QtGui.QImage.Format_RGB888)
TypeError: arguments did not match any overloaded call:
QImage(): too many arguments
QImage(QSize, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(int, int, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(bytes, int, int, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(sip.voidptr, int, int, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(bytes, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(sip.voidptr, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview'
QImage(List[str]): argument 1 has unexpected type 'memoryview'
QImage(str, format: str = None): argument 1 has unexpected type 'memoryview'
QImage(QImage): argument 1 has unexpected type 'memoryview'
QImage(Any): too many arguments
这是因为NumPy使用memoryview来优化某些任务。在这种情况下,当执行
src[:,:,::-1]
时,一种优化方式是不修改数据,而是修改访问数据的方式,这是通过
缓冲区协议来实现的。
而在这种情况下,QImage不支持这种类型的数据,所以解决方案是使用
tobytes()
或
bytes()
来访问字节。
import cv2
from PyQt5 import QtGui, QtWidgets
if __name__ == '__main__':
import sys
src = cv2.imread('image.jpg')
src = src[:,:,::-1]
h, w, ch = src.shape
bytesPerLine = ch * w
qImg = QtGui.QImage(src.data.tobytes(), w, h, bytesPerLine, QtGui.QImage.Format_RGB888)
app = QtWidgets.QApplication(sys.argv)
w = QtWidgets.QLabel()
w.setPixmap(QtGui.QPixmap.fromImage(qImg))
w.show()
sys.exit(app.exec_())
时间:
500 loops, best of 5: 523 usec per loop
另一种解决方案是使用OpenCV的
cvtColor()
函数,如果您修改了数据:
import cv2
from PyQt5 import QtGui, QtWidgets
if __name__ == '__main__':
import sys
src = cv2.imread('image.jpg')
src = cv2.cvtColor(src, cv2.COLOR_BGR2RGB)
h, w, ch = src.shape
bytesPerLine = ch * w
qImg = QtGui.QImage(src.data, w, h, bytesPerLine, QtGui.QImage.Format_RGB888)
app = QtWidgets.QApplication(sys.argv)
w = QtWidgets.QLabel()
w.setPixmap(QtGui.QPixmap.fromImage(qImg))
w.show()
sys.exit(app.exec_())
时间:
1000 loops, best of 5: 263 usec per loop
cvtColor()
将是最快的,因为毫无疑问 OpenCV 将使用 SIMD 来完成它。 - Mark Setchell