如何在 FastAPI 中將 NumPy 陣列渲染為映像：解決白方塊問題？

如何在FastAPI 中將NumPy 數組顯示為圖像

問題：

儘管有表示影像的NumPy 數組，在嘗試渲染時顯示為白色方塊。

解決方案：

選項1：以位元組形式傳回影像

使用自訂Response 類，請依照以下步驟操作：

1. 從磁碟載入映像或使用PIL 或OpenCV 將記憶體中的NumPy 陣列轉換為映像。
2. 分別使用 save() 或 imencode() 方法將影像儲存到 BytesIO 緩衝區。
3. 設定 Content-Disposition 標頭以指定內嵌檢視或下載。
4. 回傳回應中的影像位元組，其中 media_type 設定為適當的影像格式（例如「image/png」）。

選項2：以JSON 編碼的NumPy 數組形式傳回圖像

此選項不適合在瀏覽器中顯示圖像，但允許您將NumPy 數組編碼為JSON 並在客戶端將其解碼回圖像。

1. 將使用 tolist() 方法將 NumPy 陣列轉換為列表。
2. 將清單編碼為 JSON 並在回應中傳回。
3. 在客戶端，將 JSON 字串解碼回 NumPy 陣列並使用 PIL 或 OpenCV 從其建立映像。

注意：

• 對於這兩個選項，設定 Content-Disposition 標頭至關重要用於在瀏覽器中顯示影像。
• 如果 NumPy 陣列已載入到記憶體中並且可以在單一回應中有效傳回，則不需要 StreamingResponse。

以上是如何在 FastAPI 中將 NumPy 陣列渲染為映像：解決白方塊問題？的詳細內容。更多資訊請關注PHP中文網其他相關文章！