Python opencv 領域抽出 初心者

Question

OpenCVを用いて顔検出は出来たのですが、
上半身や下半身などを抽出することができないです。
顔検出できたコードのhaarcascade_frontalface_altをhaarcascade_upperbody、front_faceをupperbodyに書き換えればできると思ったのですが間違いなんでしょうか。

顔検出できたときのコードを書き換えたものです。
勉強不足かもしれませんが、よろしくお願いします。

import cv2
import matplotlib.pyplot as plt

# 画像読込み
origin_img = cv2.imread("image.jpg")
# 画像コピー
img = origin_img.copy()

# カスケードファイルのパス
cascade_path = "haarcascade_upperbody.xml"

# カスケード分類器の特徴量取得
cascade = cv2.CascadeClassifier(cascade_path)

# 画像グレースケール化
grayscale_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
# 顔検出
# minSize 最小サイズ指定
front_face_list = cascade.detectMultiScale(grayscale_img, minSize = (100, 100))

# 検出判定
print(upperbody_list)
if len(upperbody_list) == 0:
    print("Failed")
    quit()

# 検出位置描画
for (x,y,w,h) in upperbody_list:
    print("[x,y] = %d,%d [w,h] = %d,%d" %(x, y, w, h))
    cv2.rectangle(img, (x,y), (x+w, y+h), (0, 0, 255), thickness=10)
     
# 顔検出画像表示
plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
plt.show()
# 顔検出画像出力
cv2.imwrite("out.jpg", img)

# 検出画像出力
for (x,y,w,h) in upperbody_list:
    upperbody_img = origin_img[y:y+h, x:x+w]
    filename = "upperbody_" + str(x) + "-" + str(y) + ".jpg"
    cv2.imwrite(filename, upperbody_img)

このようなエラーが出てしまいました。

NameError: name 'upperbody_list' is not defined

Comments

[fana]

> 顔検出できたコードをうまく書き換えればうまくいくと思ったのですが間違いなんでしょうか。

何をどう書き換えればよいと考えているのか？が他者には不明です．
まずは思ったように書き換えてやってみると良いのでは．

[退会済みユーザー]

書き換えてやってみたものを載せておきました。

[tiitoi]

カスケード検出器の上半身検出や人検出はあまり精度がよくなかったと思います。実用的な精度でやるなら CNN 使うことになるでしょうね。

[退会済みユーザー]

そうなんですね。CNN参考にさせていただきます。

Answer 1

CNNを参考にします。

Answer 2

まず直接のエラーの原因はupperbody_listが参照だけされていて、
宣言されていないためそのような名前の変数がないと言われています。

質問文だと「front_faceをupperbodyに書き換えれば」とのことですので

# minSize 最小サイズ指定
upperbody_list = cascade.detectMultiScale(grayscale_img, minSize = (100, 100))

こうするつもりだったのですかね。

Comments

[退会済みユーザー]

そうですね。その部分を書き換えてやってみたらFailedとでてしまい、うまく検出されませんでした。