A function for draw rectangle using yolo text file on input image and save the result

draw_rectangles_from_yolo.py

import cv2

def draw_rectangles_from_yolo(image_path, text_file_path, output_image_name):
    image = cv2.imread(image_path)
    
    with open(text_file_path, 'r') as file:
        lines = file.readlines()
    
    for line in lines:
        data = line.split()
        x, y, w, h = map(int, data[1:])
        cv2.rectangle(image, (x, y), (x+w, y+h), (255, 0, 0), 2)
    
    cv2.imwrite(output_image_name, image)