画像から物体の種類、座標、幅、高さを検出する

Question

前提・実現したいこと

物体検出した画像の座標取得をしたいです。
https://kazusa-pg.com/object-detection-picture/　参考サイト

５、get_objects_information関数は、検出した物体ごとにdictionaryを生成し、
生成したdictionaryをListに加えて返します。物体が検出できないときは空のListを返します。

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

エラーメッセージは特に出ないのですが、Listが返ってこない状態です。

下記が実行コードです。

from PIL import Image
from keras_yolo3.yolo import YOLO
from objects import get_objects_information

if __name__ == '__main__':
    yolo = YOLO()
    image_path = "./picture/sample_picture.jpg"
    objects_info_list = get_objects_information(yolo, image_path)
    yolo.close_session()

    img = Image.open(image_path)
    count = 0
    for object_info in objects_info_list:
        class_name = object_info['predicted_name']
        x = object_info['x']
        y = object_info['y']
        width = object_info['width']
        height = object_info['height']
        cropped_img = img.crop((x, y, x + width, y + height))
        cropped_img.save("./picture/{}{}.jpg".format(class_name, count))
        count = count + 1

Answer 1

自己解決しました。もともと座標は取得しています。