แปลงภาพจาก PIL เป็นรูปแบบ openCV

ฉันกำลังพยายามแปลงภาพจากPILเป็นOpenCVรูปแบบ ฉันกำลังใช้OpenCV 2.4.3. นี่คือสิ่งที่ฉันพยายามจนถึงตอนนี้

>>> from PIL import Image
>>> import cv2 as cv
>>> pimg = Image.open('D:\\traffic.jpg')                           #PIL Image
>>> cimg = cv.cv.CreateImageHeader(pimg.size,cv.IPL_DEPTH_8U,3)    #CV Image
>>> cv.cv.SetData(cimg,pimg.tostring())
>>> cv.cv.NamedWindow('cimg')
>>> cv.cv.ShowImage('cimg',cimg)
>>> cv.cv.WaitKey()

แต่ฉันคิดว่ารูปภาพไม่ได้รับการแปลงเป็นรูปแบบ CV หน้าต่างแสดงภาพสีน้ำตาลขนาดใหญ่ ฉันผิดพลาดตรงไหนในการแปลงรูปภาพPILเป็นCVรูปแบบ

นอกจากนี้ทำไมฉันต้องพิมพ์cv.cvเพื่อเข้าถึงฟังก์ชัน

ใช้สิ่งนี้:

pil_image = PIL.Image.open('Image.jpg').convert('RGB') 
open_cv_image = numpy.array(pil_image) 
# Convert RGB to BGR 
open_cv_image = open_cv_image[:, :, ::-1].copy() 

นี่เป็นเวอร์ชันที่สั้นที่สุดที่ฉันสามารถหาได้โดยบันทึก / ซ่อนการแปลงเพิ่มเติม:

pil_image = PIL.Image.open('image.jpg')
opencvImage = cv2.cvtColor(numpy.array(pil_image), cv2.COLOR_RGB2BGR)

หากอ่านไฟล์จาก URL:

import cStringIO
import urllib
file = cStringIO.StringIO(urllib.urlopen(r'http://stackoverflow.com/a_nice_image.jpg').read())
pil_image = PIL.Image.open(file)
opencvImage = cv2.cvtColor(numpy.array(pil_image), cv2.COLOR_RGB2BGR)

โค้ดที่แสดงความคิดเห็นใช้งานได้เช่นกันเพียงแค่เลือกที่คุณต้องการ

import numpy as np
from PIL import Image


def convert_from_cv2_to_image(img: np.ndarray) -> Image:
    # return Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    return Image.fromarray(img)


def convert_from_image_to_cv2(img: Image) -> np.ndarray:
    # return cv2.cvtColor(numpy.array(img), cv2.COLOR_RGB2BGR)
    return np.asarray(img)