diff --git a/apps/ai/main.py b/apps/ai/main.py index ef9a38f8..0d11d03c 100644 --- a/apps/ai/main.py +++ b/apps/ai/main.py @@ -110,6 +110,13 @@ def draw(path, results): font = cv2.FONT_HERSHEY_COMPLEX_SMALL for i in results: for m in results[i]: - cv2.rectangle(img, results[i][m][0], results[i][m][1], (0, 255, 0), 4) - cv2.putText(img, i, results[i][m][0], font, 2, (0, 0, 255), 1) + cv2.rectangle(img, m[0], m[1], (0, 255, 0), 4) + cv2.putText(img, i, m[0], font, 2, (0, 0, 255), 1) cv2.imwrite(path, img) + +{'qiping': [ + [ + (1142.2359619140625, 1295.6361083984375), (1363.11669921875, 1347.201904296875) + ] + ] + } \ No newline at end of file