如何將以字符串形式存儲的圖像加載到numpy數組中？

Question

有幾個將圖像文件加載到numpy數組中的示例 ，但是本地文件系統上的文件中沒有圖像。 相反，我有一個字符串變量img ，其中包含構成圖像的數據。 例如：

# Currently here:
with open("imageFile.png") as f:
    img = f.read()

# Would like to accomplish this:
npArray = loadStringToArray(img)

將此文件寫回到磁盤並調用更常見的加載函數之一似乎是不切實際的，因此，如何將“字符串圖像”加載到numpy數組中？

有點背景-在我的應用程序中，此字符串圖像是通過python套接字從客戶端傳遞的。 該映像不在本地計算機上； 上面的例子就是這樣-一個例子。 雖然可以將文件寫入本地文件系統， 然后再讀回，但我想避免這樣做。

Answer 1

我認為您想要做的就是將字符串轉換為numpy數組，因為您已經將圖像作為字符串。

您可以將numpy.fromstring（）用於該任務。

例子：-

np.fromstring('\\x01\\x02', dtype=np.uint8) ----->數組（[1，2]，dtype = uint8）

np.fromstring('1 2', dtype=int, sep=' ') -----> array（[1，2]）

np.fromstring('1, 2', dtype=int, sep=',') -----> array（[1，2]）

np.fromstring('\\x01\\x02\\x03\\x04\\x05', dtype=np.uint8, count=3) ----->數組（[1,2,3]，dtype = uint8）

Answer 2

無需BytesIO ， io模塊提供“文件狀”對象，尤其是StringIO和BytesIO ，它們為內存中對象（如字符串）配備了文件狀接口。 許多期望文件參數的函數會很樂意接受StringIO或BytesIO 。