Browse Source

Add files via upload

tags/v1.0.0
Alexey Mikhailyuk 4 years ago
parent
commit
a058678b8b
No account linked to committer's email address
1 changed files with 1 additions and 2 deletions
  1. 1
    2
      test.py

+ 1
- 2
test.py View File

parser.add_argument('-t', help='Yolo text file', dest='txt',required=True) parser.add_argument('-t', help='Yolo text file', dest='txt',required=True)
args = parser.parse_args() args = parser.parse_args()
#напишите путь к нужному вам изображению
frame = cv2.imread(args.img) frame = cv2.imread(args.img)
coordinates = [] coordinates = []
with open(args.txt, "r") as lines: with open(args.txt, "r") as lines:
print(x1,y1,w2,h2) print(x1,y1,w2,h2)
w, h = int(w2 * wT), int(h2 * hT) w, h = int(w2 * wT), int(h2 * hT)
x, y = int((x1 * wT) - w / 2), int((y1 * hT) - h / 2) x, y = int((x1 * wT) - w / 2), int((y1 * hT) - h / 2)
# # 0 1 2 3 x1 - центр по x в процентах y1- центр по y в процентах w1- центр по w в процентах h1- центр по h в процентах
# # 0 1 2 3 x1 - центр по x в процентах y1- центр по y в процентах w2- центр по w в процентах h2- центр по h в процентах
cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,255)) cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,255))
cv2.putText(frame,str(idx), (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (128, 0, 255), 2) cv2.putText(frame,str(idx), (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (128, 0, 255), 2)
cv2.imshow("f", frame) cv2.imshow("f", frame)

Loading…
Cancel
Save