先贴代码,周末再补充解析。
这个篇章主要是对标注好的标签进行可视化,虽然比较简单,但是可以从可视化代码中学习到YOLOv8是如何对标签进行解析的。
import cv2
import numpy as np
import os
import random
def read_pose_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])
x_center = float(parts[1])
y_center = float(parts[2])
width = float(parts[3])
height = float(parts[4])
poses=np.array(parts[5:]).reshape(-1,3).astype(np.float32)
filtered_poses = [pose[:2] f