You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test.py 1.0KB

1234567891011121314151617181920212223
  1. import cv2
  2. import time
  3. import argparse
  4. parser = argparse.ArgumentParser(description='Test yolo data.')
  5. parser.add_argument('-i', help='Image', dest='img', required=True)
  6. parser.add_argument('-t', help='Yolo text file', dest='txt',required=True)
  7. args = parser.parse_args()
  8. #напишите путь к нужному вам изображению
  9. frame = cv2.imread(args.img)
  10. coordinates = []
  11. with open(args.txt, "r") as lines:
  12. for line in lines:
  13. hT, wT, cT = frame.shape
  14. coordinates = line[2:].rstrip('\n').split(' ')
  15. x1, y1,w2,h2 = float(coordinates[0]), float(coordinates[1]), float(coordinates[2]), float(coordinates[3])
  16. print(x1,y1,w2,h2)
  17. w, h = int(w2 * wT), int(h2 * hT)
  18. x, y = int((x1 * wT) - w / 2), int((y1 * hT) - h / 2)
  19. # # 0 1 2 3 x1 - центр по x в процентах y1- центр по y в процентах w1- центр по w в процентах h1- центр по h в процентах
  20. cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,255))
  21. cv2.imshow("f", frame)
  22. cv2.waitKey(0)