異なるクラスの同時検出

Question

前提

python3.8でyolov5を使用しています

実現したいこと

たとえば、条件分岐で写真に人とスーツケースが写っていたらバウンディングボックスを出力するのようなことをしたいと考えています。しかし、どうやら一度の認識で一個の物体しか認識していない？ようで同時に２つ以上のクラス検出ができません。
どのように解決したらよろしいでしょうか。
よろしくお願いいたします。

該当のソースコード

class YOLOV5:
    def yolov5(self):
        model_name='yolov5s.pt'
        model = torch.hub.load(os.getcwd()+'/YOLOv5/yolov5/', 'custom', source='local', path = model_name, force_reload = True)
        color = (255,0,0) 
        RGB_image = cv2.imread("/home/limlab/YOLOv5/yolov5/test8.jpg")
        results = model(RGB_image)  # 画像パスを設定し、物体検出を行う

        for *box, conf, cls in results.xyxy[0]:  
            if conf > 0.5:
                s = model.names[int(cls)]
                print(s)
                if s == 'person' and s == "suitcase":
                    #--- 枠描画
                    cv2.rectangle(RGB_image,(int(box[0]), int(box[1])),(int(box[2]), int(box[3])),color,thickness=2,)
                    #--- 文字枠と文字列描画
                    cv2.rectangle(RGB_image, (int(box[0]), int(box[1])-20), (int(box[0])+len(s)*10, int(box[1])), color, -1)
                    cv2.putText(RGB_image, s, (int(box[0]), int(box[1])-5), cv2.FONT_HERSHEY_PLAIN, 1, [225,255,255], 1, cv2.LINE_AA)
            cv2.imshow('color',RGB_image)
            if cv2.waitKey(0) & 0xff == 27:#ESCで終了
                cv2.destroyAllWindows()
                    

出力結果

person
person
suitcase
person
person

Comments

[jbpb0]

たとえば、

num_person = 0
num_suitcase = 0
for *box, conf, cls in results.xyxy[0]:
if conf > 0.5:
s = model.names[int(cls)]
print(s)
if s == 'person':
num_person = num_person + 1
if s == "suitcase":
num_suitcase = num_suitcase + 1

みたいに、「人」と「スーツケース」の数を数えます
(上記はインデントが消えて見えるので、インデントをちゃんと入れてください)

次に、上記で数えた結果で、
「num_person」が「0」よりも大きい
かつ
「num_suitcase」が「0」よりも大きい
場合のみ、質問のコードの

> for *box, conf, cls in results.xyxy[0]:

以下に進みます

なお、質問のコードから、can110さんの回答のように修正します

上記のようにしたら、どうでしょうか？

[reiya_123]

ご返信ありがとうございます。
無事解決することができました。
ありがとうございました。

Answer 1

        num_person = 0
        num_suitcase = 0
        for *box, conf, cls in results.xyxy[0]:  
            if conf > 0.5:
                s = model.names[int(cls)]
                print(s)
                if s == 'person':
                    num_person = num_person + 1
                if s == "suitcase":
                    num_suitcase = num_suitcase + 1

みたいにして、「人」と「スーツケース」の数を数えます

　
次に、上記で数えた結果で、
「num_person」が「0」よりも大きい
かつ
「num_suitcase」が「0」よりも大きい
場合のみ、質問のコードの

        for *box, conf, cls in results.xyxy[0]: 

以下を実行します

　
なお、質問のコードから、can110さんの回答のように修正します

　
上記のようにしたら、

写真に人とスーツケースが写っていたらバウンディングボックスを出力する

ができると思います

Answer 2

yolov5については分かりませんが提示コードにおいて
if s == 'person' and s == "suitcase":は同時に成立することはありません。
if s == 'person' or s == "suitcase":の誤りかと思います。