先貼代碼,后面再補充解析。
這個篇章主要是對標注好的標簽進行可視化,雖然比較簡單,但是可以從可視化代碼中學習到YOLOv8是如何對標簽進行解析的。
下面直接貼代碼:
import cv2
import numpy as np
import osdef read_det_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])labels.append((class_id, x_center, y_center, width, height))return labelsdef det_label(label_file,img_file,save_file):img=cv2.imread(img_file)image=img.copy()img_height,img_width,_=image.s