vladislav.kan/OpenCV
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

изменно на процентное соотношение

Browse Source
This commit is contained in:
Vlad 2024-02-01 16:33:49 +06:00
parent b2a8f10092
commit a93054997e
1 changed files with 49 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
text2.py
View File

@ -7,50 +7,65 @@ pytesseract.pytesseract.tesseract_cmd = r'C:\Users\ASUS\Desktop\fishrungame\trea
def draw_boxes(image, boxes, texts): def draw_boxes(image, boxes, texts):
for i, box in enumerate(boxes): for i, box in enumerate(boxes):
x, y, x_w, y_h = box x, y, x_w, y_h = box
cv2.rectangle(image, (x, y), (x + x_w, y + y_h), (0, 255, 0), 2) cv2.rectangle(image, (x, y), (x + x_w, y + y_h), (0, 255, 0), 2)
cv2.putText(image, texts[i], (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2) cv2.putText(image, texts[i], (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
def process_image(image_path, coordinates_list): def process_image(image_path, percentage_coordinates_list):
image = cv2.imread(image_path) # Загрузка изображения и изменение размера
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) original_image = cv2.imread(image_path)
resized_image = cv2.resize(original_image, (662, 429))
# Применение OCR # Преобразование в оттенки серого
custom_config = r'--oem 3 --psm 6 -l kir+eng+rus' gray = cv2.cvtColor(resized_image, cv2.COLOR_BGR2GRAY)
text = pytesseract.image_to_string(gray, config=custom_config)
# Получение координат и размеров квадратов # Применение OCR
h, w, _ = image.shape custom_config = r'--oem 3 --psm 6 -l kir+eng+rus'
boxes = [(int(x), int(h - y_h), int(x_w - x), int(h - y)) for x, y, x_w, y_h in coordinates_list] text = pytesseract.image_to_string(gray, config=custom_config)
# Фильтрация квадратов по проценту (если нужно) # Получение размеров изображения
percentage_threshold = 1 h, w, _ = resized_image.shape
filtered_boxes = [box for box in boxes if (box[2] * box[3]) / (w * h) * 100 >= percentage_threshold]
# Извлечение текста из области каждого квадрата # Преобразование процентных координат в абсолютные
texts = [] coordinates_list = [
for box in filtered_boxes: (
x, y, x_w, y_h = box int(x * w / 100),
crop_img = gray[y:y + y_h, x:x + x_w] int(y * h / 100),
cropped_text = pytesseract.image_to_string(crop_img, config=custom_config) int(x_w * w / 100),
texts.append(cropped_text) int(y_h * h / 100)
print(f"Text inside the box at ({x}, {y}): {cropped_text}") )
for x, y, x_w, y_h in percentage_coordinates_list
]
# Рисование квадратов и вывод текста # Фильтрация квадратов по проценту (если нужно)
draw_boxes(image, filtered_boxes, texts) percentage_threshold = 1
filtered_boxes = [box for box in coordinates_list if (box[2] * box[3]) / (w * h) * 100 >= percentage_threshold]
# Отображение изображения с квадратами с использованием matplotlib # Извлечение текста из области каждого квадрата
plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)) texts = []
plt.title('Image with Boxes and Text') for box in filtered_boxes:
plt.show() x, y, x_w, y_h = box
crop_img = gray[y:y + y_h, x:x + x_w]
cropped_text = pytesseract.image_to_string(crop_img, config=custom_config)
texts.append(cropped_text)
print(f"Text inside the box at ({x}, {y}): {cropped_text}")
# Рисование квадратов и вывод текста
draw_boxes(resized_image, filtered_boxes, texts)
# Отображение изображения с квадратами с использованием matplotlib
plt.imshow(cv2.cvtColor(resized_image, cv2.COLOR_BGR2RGB))
plt.title('Image with Boxes and Text')
plt.show()
if __name__ == "__main__": if __name__ == "__main__":
image_path = 'img/img.png' image_path = 'img/img.png'
# Задайте координаты для каждого квадрата в формате (x, y, x_w, y_h) # Задайте координаты для каждого квадрата в процентном соотношении (x, y, x_w, y_h)
coordinates_list = [(248, 399, 480, 330), (248, 399, 480, 282), (248, 399, 480, 230), (282, 399, 480, 190), (248, 399, 600, 160), (248, 399, 430, 120), (452, 399, 600, 103), (450, 395, 600, 60)] percentage_coordinates_list = [(38, 23, 30, 11), (38, 35.5, 30, 10.5), (38, 48, 30, 7), (38, 56.4, 30, 5.5),
(38, 65, 45, 5.5), (38, 73, 20, 6), (68, 75.1, 20, 6.3), (68, 87, 20, 6)]
process_image(image_path, coordinates_list) process_image(image_path, percentage_coordinates_list)