為什么我得到 'numpy.ndarray' 對象沒有屬性 'convert'？

Question

我目前在做一個機器學習項目，出於某種原因，我自己實現了數據增強。

圖像的類型是

'PIL.JpegImagePlugin.JpegImageFile'

我想將我的圖像附加到名為圖像的列表中。

#method "scale_augmentation" is self defined function. Return image.
f_processed = self.scale_augmentation(image = f) 

#error occurs here                
self.images.append(Image.fromarray(f_processed.convert('RGB'), dtype=np.float32) / 255.)

我被這個錯誤困擾了這么久，我已經解決了很多關於這個錯誤的堆棧溢出問題。 提前致謝。

---

Scale_augmentation 函數看起來像這樣

def scale_augmentation(image, scale_range=(256, 400), crop_size=224):
scale_size = np.random.randint(*scale_range)
image = imresize(image, (scale_size, scale_size))
image = random_crop(image, (crop_size, crop_size))
return image

---

也錯誤看起來像這樣

--> 100 self.images.append(Image.fromarray(f_processed.convert('RGB'), dtype=np.float32) / 255.)
AttributeError: 'numpy.ndarray' object has no attribute 'convert'

Answer 1

首先通過Image.fromarray將其轉換為圖像對象，然后使用convert

嘗試這個：

self.images.append((Image.fromarray(f_processed, dtype=np.float32).convert('RGB') / 255.)

明確的步驟是：

image = Image.fromarray(image)

image = image.convert("RGB")

現在圖像是 jpeg 格式，你可以：

image.save('test_test.jpeg')