前提・実現したいこと
YOLOを用いて認識を行ったあとに、その物体の位置情報を用いて
デフォルメされた別の画像を作成しようとしています。
pythonのコーディングの初心者なのとjsonファイルというものを初めて触るので
どのように読み込んだらよいか分かりません。
もしお時間あればその先の部分も
自分がやりたいことを表現してみたはいいものの
文法的に合っていないと思われるので訂正していただけると幸いです。
発生している問題・エラーメッセージ
20行目の
print(result)
では
[{'label': 'car', 'confidence': 0.84797806, 'topleft': {'x': 80, 'y': 113},
'bottomright': {'x': 555, 'y': 467}}, {'label': 'truck', 'confidence': 0.8014206
, 'topleft': {'x': 462, 'y': 81}, 'bottomright': {'x': 694, 'y': 167}}]
と出力されます。
26行目でこれを構造体のように扱いたいのですがjson.loadではエラーを吐きます。
print(result[0])で
[{'label': 'car', 'confidence': 0.84797806, 'topleft': {'x': 80, 'y': 113},'bottomright': {'x': 555, 'y': 467}}]
にアクセスはできたのですが、labelの情報をif文の条件に使い、x,yの座標を適宜呼び出す方法が分かりません。
該当のソースコード
from darkflow.net.build import TFNet
import cv2
import numpy as np
import json
print(result)
% 作成したい画像の下地の画像の読み込み
bImg = cv2.imread("../../image/road1.jpg")
bSize = tuple([bImg.shape[1], bImg.shape[0]])
ob_dict = json.load(result)
for x in len(result):
% 各ラベルに対応した出力の画像の取得 if result(x) == 'car': pngImage = cv2.imread("../../image/toyota_back2.png" , cv2.IMREAD_UNCHANGED) elif result(x) == 'person': pngImage = cv2.imread("../../image/man.png" , cv2.IMREAD_UNCHANGED) % 物体を描画したときの倍率の計算 scale = (ob_dict(bottomright,x) - ob_dect(topleft,x)) * (ob_dict(bottomright,y) - ob_dect(topleft,y)) / (bSize[0] * bsize[1]) % 画像の書き込み resultImage = transparentOverlay(bImg,pngImage,(ob_dect(bottomright,x),ob_dect(bottomright,y)),scale)
補足情報(FW/ツールのバージョンなど)
yoloはdarkflowでtensorflowベースで動かしています。
回答1件
あなたの回答
tips
プレビュー