先贴代码,周末再补充解析。
这个篇章主要是对标注好的标签进行可视化,虽然比较简单,但是可以从可视化代码中学习到YOLOv8是如何对标签进行解析的。
import cv2
import numpy as np
import os
import randomdef read_obb_labels(label_file_path):with open(label_file_path, 'r') as file:lines = file.readlines()labels = []for line in lines:parts = line.strip().split()class_id = int(parts[0])points=np.array(parts[1:]).reshape(-1,2).astype(np.float32)labels.append((class_id,points))return labelsdef obb_label(label_file,img_file,save_file):img=cv2.imread(img_file)image=img.copy()img_height,img_wid