Pythonを用いて写真画像内にある物体外側の輪郭の描画を実現したいのですが、画像を保存できません。

Question

前提・実現したいこと

Pythonを用いて写真画像内にある物体外側の輪郭の描画を実現したいと思っております。
実行したところ、画像を保存する行でエラーが発生してしまいます。
どうかご教授願います。

発生している問題・エラーメッセージ

    img.save('edited_img.jpg')
AttributeError: 'numpy.ndarray' object has no attribute 'save'

該当のソースコード

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
import cv2

im = cv2.imread('test.jpg')
imgray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
ret,thresh = cv2.threshold(imgray,127,255,cv2.THRESH_OTSU)
contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)


external_contours = np.zeros(im.shape)
 
for i in range(len(contours)):
    
    if hierarchy[0][i][3] == -1:
        
        img = cv2.drawContours(external_contours, contours, i, 255, -1)

img.save('edited_img.jpg')

Answer 1

save メソッドがあるのは、Pillow というライブラリの PIL.Image オブジェクトです。
OpenCV は画像は numpy 配列として扱うので、cv2.imwrite() で保存してください。

- img.save('edited_img.jpg')
+ cv2.imwrite("edited_img.jpg", img)

Comments

[ponpon0987]

無事保存の問題は解決できました！
ありがとうございます！