Cách đọc hình ảnh từ bộ nhớ đệm (StringIO) hoặc từ url bằng thư viện python opencv

Question

Chỉ cần chia sẻ cách tạo đối tượng hình ảnh opencv từ bộ nhớ đệm hoặc từ url để cải thiện hiệu suất.

Đôi khi chúng tôi nhận được hình ảnh nhị phân từ url, để tránh thêm tệp IO, chúng tôi muốn imread hình ảnh này từ bộ nhớ đệm hoặc từ url, nhưng chỉ đọc hỗ trợ đọc hình ảnh từ hệ thống tệp có đường dẫn.

Answer 1

Để tạo một đối tượng hình ảnh OpenCV với trong bộ nhớ đệm (StringIO), chúng ta có thể sử dụng OpenCV API imdecode, xem mã bên dưới:

import cv2 
import numpy as np 
from urllib2 import urlopen 
from cStringIO import StringIO 

def create_opencv_image_from_stringio(img_stream, cv2_img_flag=0): 
    img_stream.seek(0) 
    img_array = np.asarray(bytearray(img_stream.read()), dtype=np.uint8) 
    return cv2.imdecode(img_array, cv2_img_flag) 

def create_opencv_image_from_url(url, cv2_img_flag=0): 
    request = urlopen(url) 
    img_array = np.asarray(bytearray(request.read()), dtype=np.uint8) 
    return cv2.imdecode(img_array, cv2_img_flag)