使用 OpenCV 不存檔直接讀入 Flask 所傳進的圖片

| | |

在實做 flask 時，我透過 curl 指令取得了張圖片，但它的資料格式是 'werkzeug.datastructures.FileStorage'，不能直接用 cv2.imread ，但我又不想跟範例程式碼一樣，先存出再讀回。所以找了其他的方法來實做。

先存出再讀回

一般常見的範例都會先存出再讀回，將像下面程式碼一樣。

1
2
3
4
5
data =request.files['file']
filename = secure_filename(file.filename) 
filepath = os.path.join("/tmp", filename);
file.save(filepath)
img = cv2.imread(filepath)

直接從記憶體中讀取

不過要先存出再讀回，我實在不是很喜歡，所以找到的 cv2.imdecode，根據解釋：

cv2.imdecode()
函數從指定的記憶體快取中讀取數據，並把數據解碼成圖像格式；主要用於從網路傳輸數據中恢復出圖像。


看到網路傳輸數據眼睛就亮起來了有沒有！當下就決定把 cv.imread 換成 cv2.imdecod，並搭配 numpy 的fromstring 使用字串來建立一個矩陣，完整程式碼如下：

1
2
3
filestr = request.files['file'].read()
npimg = numpy.fromstring(filestr, numpy.uint8)
img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)

資料型態

用這方式讀進來的資料型態是 uint8，在我的應用中，我會接著喂進 TensorFlow 做 predict，結果會跳出：

TensorFlow TypeError
Value passed to parameter input has DataType uint8 not in list of allowed values: float16, float32


所以我又在最後加了行型態轉換：

1
2
3
4
filestr = request.files['file'].read()
npimg = numpy.fromstring(filestr, numpy.uint8)
img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)
img = img.astype("float32")



關於 IO 錯誤描述
之前文句中出現 省下那兩次的 IO 的敘述有誤，感謝 Chi Gas 的指正。
在使用 imdecode() 和 imencode() 也是進行了 IO ，所以用省下 IO 的描述並不正確。

參考資料

  1. -牧野- (2018-01-25)。OpenCV-Python cv2.imdecode()和cv2.imencode() 图片解码和编码 。檢自 牧野的博客-CSDN博客(2020-02-19)。
  2. flamelite (2017-12-28)。Reading image file (file storage object) using CV2 。檢自 Stack Overflow (2020-02-19)。
  3. tbicr (2013-11-16)。Read file data without saving it in Flask 。檢自 Stack Overflow (2020-02-19)。

更新紀錄

最後更新日期： 2020-08-19
  • 2020-08-19 更新 修正錯誤描述
  • 2020-02-19 發布