[python][原创]关于DOAT数据集中hbb和obb两种标注区别

import cv2
import os
import numpy as np

image_file = r'C:\Users\fut\Desktop\extract\P2750.png'
hbb_file = r'C:\Users\fut\Desktop\extract\P2750_hbb.txt'
obb_file = r'C:\Users\fut\Desktop\extract\P2750_obb.txt'


def load_lines(file):
    with open(file, 'r') as f:
        lines = f.read().rstrip('\n').split('\n')
    return lines[2:]


def draw_image(frame, lines):
    for line in lines:
        data = line.split(' ')
        x1 = int(float(data[0]))
        y1 = int(float(data[1]))
        x2 = int(float(data[2]))
        y2 = int(float(data[3]))
        x3 = int(float(data[4]))
        y3 = int(float(data[5]))
        x4 = int(float(data[6]))
        y4 = int(float(data[7]))
        name = data[8]
        box = np.array([[x1, y1], [x2, y2], [x3, y3], [x4, y4]], np.int32)
        box = box.reshape((-1, 1, 2))
        cv2.polylines(frame, [box], True, (0, 0, 255), 2)
    return frame


hbb_lines = load_lines(hbb_file)
obb_lines = load_lines(obb_file)
hbb_img = draw_image(cv2.imread(image_file), hbb_lines)
obb_img = draw_image(cv2.imread(image_file), obb_lines)
cv2.imshow('hbb', hbb_img)
cv2.imshow('obb', obb_img)
cv2.waitKey(0)