wulin/rknn-toolkit2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update test.py

Browse Source
This commit is contained in:
shaoshengsong 2022-10-18 19:07:46 +08:00 committed by GitHub
parent d8074d7954
commit 86174b03fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
examples/onnx/yolov5/test.py
View File

@ -184,7 +184,6 @@ def yolov5_post_process(input_data):
def draw(image, boxes, scores, classes): def draw(image, boxes, scores, classes):
"""Draw the boxes on the image. """Draw the boxes on the image.
# Argument: # Argument:
image: original image. image: original image.
boxes: ndarray, boxes of objects. boxes: ndarray, boxes of objects.
@ -193,17 +192,17 @@ def draw(image, boxes, scores, classes):
all_classes: all classes name. all_classes: all classes name.
""" """
for box, score, cl in zip(boxes, scores, classes): for box, score, cl in zip(boxes, scores, classes):
top, left, right, bottom = box left, top, right, bottom = box
print('class: {}, score: {}'.format(CLASSES[cl], score)) print('class: {}, score: {}'.format(CLASSES[cl], score))
print('box coordinate left,top,right,down: [{}, {}, {}, {}]'.format(top, left, right, bottom)) print('box coordinate left,top,right,bottom: [{}, {}, {}, {}]'.format(left, top, right, bottom))
top = int(top)
left = int(left) left = int(left)
top = int(top)
right = int(right) right = int(right)
bottom = int(bottom) bottom = int(bottom)
cv2.rectangle(image, (top, left), (right, bottom), (255, 0, 0), 2) cv2.rectangle(image, (left, top), (right, bottom), (255, 0, 0), 2)
cv2.putText(image, '{0} {1:.2f}'.format(CLASSES[cl], score), cv2.putText(image, '{0} {1:.2f}'.format(CLASSES[cl], score),
(top, left - 6), (left, top - 6),
cv2.FONT_HERSHEY_SIMPLEX, cv2.FONT_HERSHEY_SIMPLEX,
0.6, (0, 0, 255), 2) 0.6, (0, 0, 255), 2)