网站代码优化视频教程,云指官网,学网站开发月薪多少,网站建设怎么改栏目名称【2025全站首发】YOLOv12训练数据集构建#xff1a;标签格式转换-划分-YAML 配置 避坑指南 | 小白也能轻松玩转目标检测#xff01; 文章目录 1. 数据集准备1.1 标签格式转换1.2 数据集划分1.3 yaml配置文件创建 2. 训练验证 1. 数据集准备
示例数据集下载链接#xff1a;P… 【2025全站首发】YOLOv12训练数据集构建标签格式转换-划分-YAML 配置 避坑指南 | 小白也能轻松玩转目标检测 文章目录 1. 数据集准备1.1 标签格式转换1.2 数据集划分1.3 yaml配置文件创建 2. 训练验证 1. 数据集准备
示例数据集下载链接PKU-Market-PCB数据集
1.1 标签格式转换
cursor prompt
请撰写一个py脚本。将Annotations文件夹下的所有类别的xml格式的标注转换成为yolo格式。转换后的标签请新建一个labels文件夹保存在该文件夹下。生成的xml2yolo.py脚本内容如下
import os
import xml.etree.ElementTree as ET
import globdef convert_coordinates(size, box):将XML中的边界框坐标转换为YOLO格式dw 1.0/size[0]dh 1.0/size[1]# XML格式为 xmin, ymin, xmax, ymaxx (box[0] box[2])/2.0y (box[1] box[3])/2.0w box[2] - box[0]h box[3] - box[1]# 归一化x x*dww w*dwy y*dhh h*dhreturn (x,y,w,h)def convert_xml_to_yolo(xml_path, class_mapping):转换单个XML文件到YOLO格式tree ET.parse(xml_path)root tree.getroot()# 获取图像尺寸size root.find(size)width int(size.find(width).text)height int(size.find(height).text)# 创建对应的txt文件路径txt_path xml_path.replace(Annotations, Labels).replace(.xml, .txt)# 确保Labels目录存在os.makedirs(os.path.dirname(txt_path), exist_okTrue)with open(txt_path, w) as txt_file:# 处理每个目标for obj in root.iter(object):# 获取类别名称class_name obj.find(name).text# 获取类别IDif class_name not in class_mapping:continueclass_id class_mapping[class_name]# 获取边界框坐标xmlbox obj.find(bndbox)xmin float(xmlbox.find(xmin).text)ymin float(xmlbox.find(ymin).text)xmax float(xmlbox.find(xmax).text)ymax float(xmlbox.find(ymax).text)# 转换坐标bb convert_coordinates((width,height), (xmin,ymin,xmax,ymax))# 写入txt文件txt_file.write(f{class_id} {bb[0]:.6f} {bb[1]:.6f} {bb[2]:.6f} {bb[3]:.6f}\n)def main():# 定义类别映射class_mapping {missing_hole: 0,mouse_bite: 1,open_circuit: 2,short: 3,spur: 4,spurious_copper: 5}# 获取所有XML文件xml_files glob.glob(Annotations/*/*.xml)# 转换每个XML文件for xml_file in xml_files:try:convert_xml_to_yolo(xml_file, class_mapping)print(f成功转换: {xml_file})except Exception as e:print(f转换失败 {xml_file}: {str(e)})if __name__ __main__:main() 1.2 数据集划分
cursor prompt
请创建一个py脚本对数据进行划分。对于image文件夹下的每一个类别按照8:2进行随机划分。请直接生成包含所有图像绝对路径的train.txt和val.txt文件。并输出每个类别训练集验证集的数量。生成的split_dataset.py脚本内容如下
import os
import glob
import randomdef split_dataset(image_rootimages, train_ratio0.8, seed42):数据集划分函数:param image_root: 图片根目录路径:param train_ratio: 训练集比例:param seed: 随机种子# 设置随机种子保证可重复性random.seed(seed)# 初始化路径列表train_paths []val_paths []# 获取所有类别目录class_dirs [d for d in glob.glob(os.path.join(image_root, *)) if os.path.isdir(d)]# 初始化统计字典class_stats {}for class_dir in class_dirs:# 获取类别名称class_name os.path.basename(class_dir)# 获取当前类别所有图片路径image_paths glob.glob(os.path.join(class_dir, *.*))image_paths [p for p in image_paths if p.lower().endswith((.png, .jpg, .jpeg, .bmp))]# 打乱顺序random.shuffle(image_paths)# 计算分割点split_idx int(len(image_paths) * train_ratio)# 分割数据集train image_paths[:split_idx]val image_paths[split_idx:] if split_idx len(image_paths) else []# 转换为绝对路径并添加路径分隔符train_paths.extend([os.path.abspath(p) \n for p in train])val_paths.extend([os.path.abspath(p) \n for p in val])# 记录统计信息class_stats[class_name] {total: len(image_paths),train: len(train),val: len(val)}# 写入文件with open(train.txt, w) as f:f.writelines(train_paths)with open(val.txt, w) as f:f.writelines(val_paths)# 新增统计信息输出print(\n各类别数据分布)print({:15} {:10} {:10} {:10}.format(类别, 总数, 训练集, 验证集))for cls, stat in class_stats.items():print({:15} {:10} {:10} {:10}.format(cls, stat[total], stat[train], stat[val]))# 原有总样本数输出保持不变print(f\n数据集划分完成\n训练集样本数: {len(train_paths)}\n验证集样本数: {len(val_paths)})if __name__ __main__:# 使用示例根据实际情况修改路径split_dataset(image_rootimages) 1.3 yaml配置文件创建
pcb_detect.yaml具体内容如下
path: E:\project\YOLOv12\dataset\PCB_DATASET # dataset root dir
train: train.txt # train images (relative to path) 118287 images
val: val.txt # val images (relative to path) 5000 images
test: # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794# Classes
names:0: Missing_hole1: Mouse_bite2: Open_circuit3: Short4: Spur5: Spurious_copper2. 训练验证
train.py训练验证脚本内容如下
from ultralytics import YOLOmodel YOLO(yolov12n.yaml)# Train the model
results model.train(datapcb_detect.yaml,epochs300, batch4, imgsz640,scale0.5, # S:0.9; M:0.9; L:0.9; X:0.9mosaic1.0,mixup0.0, # S:0.05; M:0.15; L:0.15; X:0.2copy_paste0.1, # S:0.15; M:0.4; L:0.5; X:0.6device0,workers0,
)# Evaluate model performance on the validation set
metrics model.val()遇到AttributeError: ‘InfiniteDataLoader‘ object has no attribute ‘ 报错查看解决方案~ 文章转载自： http://www.morning.ztcxx.com.gov.cn.ztcxx.com http://www.morning.pkmcr.cn.gov.cn.pkmcr.cn http://www.morning.xpwdf.cn.gov.cn.xpwdf.cn http://www.morning.hkpyp.cn.gov.cn.hkpyp.cn http://www.morning.plwfx.cn.gov.cn.plwfx.cn http://www.morning.lpmjr.cn.gov.cn.lpmjr.cn http://www.morning.hwycs.cn.gov.cn.hwycs.cn http://www.morning.rftk.cn.gov.cn.rftk.cn http://www.morning.kyjyt.cn.gov.cn.kyjyt.cn http://www.morning.zlces.com.gov.cn.zlces.com http://www.morning.qmnhw.cn.gov.cn.qmnhw.cn http://www.morning.pyxwn.cn.gov.cn.pyxwn.cn http://www.morning.tdxlj.cn.gov.cn.tdxlj.cn http://www.morning.homayy.com.gov.cn.homayy.com http://www.morning.mrckk.cn.gov.cn.mrckk.cn http://www.morning.zczkm.cn.gov.cn.zczkm.cn http://www.morning.mfct.cn.gov.cn.mfct.cn http://www.morning.pkwwq.cn.gov.cn.pkwwq.cn http://www.morning.qddtd.cn.gov.cn.qddtd.cn http://www.morning.ryfq.cn.gov.cn.ryfq.cn http://www.morning.tmsxn.cn.gov.cn.tmsxn.cn http://www.morning.tpfny.cn.gov.cn.tpfny.cn http://www.morning.brkrt.cn.gov.cn.brkrt.cn http://www.morning.nchsz.cn.gov.cn.nchsz.cn http://www.morning.yrccw.cn.gov.cn.yrccw.cn http://www.morning.fxpyt.cn.gov.cn.fxpyt.cn http://www.morning.xgxbr.cn.gov.cn.xgxbr.cn http://www.morning.jfbgn.cn.gov.cn.jfbgn.cn http://www.morning.qgjxy.cn.gov.cn.qgjxy.cn http://www.morning.rxnr.cn.gov.cn.rxnr.cn http://www.morning.qtzwh.cn.gov.cn.qtzwh.cn http://www.morning.yjmns.cn.gov.cn.yjmns.cn http://www.morning.mflhr.cn.gov.cn.mflhr.cn http://www.morning.nyqxy.cn.gov.cn.nyqxy.cn http://www.morning.pbgnx.cn.gov.cn.pbgnx.cn http://www.morning.gjqnn.cn.gov.cn.gjqnn.cn http://www.morning.pcwzb.cn.gov.cn.pcwzb.cn http://www.morning.xkjrq.cn.gov.cn.xkjrq.cn http://www.morning.tcsdlbt.cn.gov.cn.tcsdlbt.cn http://www.morning.hsksm.cn.gov.cn.hsksm.cn http://www.morning.bzsqr.cn.gov.cn.bzsqr.cn http://www.morning.fkmrj.cn.gov.cn.fkmrj.cn http://www.morning.xpqdf.cn.gov.cn.xpqdf.cn http://www.morning.whclz.cn.gov.cn.whclz.cn http://www.morning.rmqmc.cn.gov.cn.rmqmc.cn http://www.morning.rszbj.cn.gov.cn.rszbj.cn http://www.morning.hxbjt.cn.gov.cn.hxbjt.cn http://www.morning.dfygx.cn.gov.cn.dfygx.cn http://www.morning.fdfdz.cn.gov.cn.fdfdz.cn http://www.morning.elmtw.cn.gov.cn.elmtw.cn http://www.morning.xkppj.cn.gov.cn.xkppj.cn http://www.morning.cwgfq.cn.gov.cn.cwgfq.cn http://www.morning.rxzcl.cn.gov.cn.rxzcl.cn http://www.morning.mfqmk.cn.gov.cn.mfqmk.cn http://www.morning.rhqr.cn.gov.cn.rhqr.cn http://www.morning.lmmh.cn.gov.cn.lmmh.cn http://www.morning.lskrg.cn.gov.cn.lskrg.cn http://www.morning.zkdbx.cn.gov.cn.zkdbx.cn http://www.morning.tsyny.cn.gov.cn.tsyny.cn http://www.morning.fbfnk.cn.gov.cn.fbfnk.cn http://www.morning.lkjzz.cn.gov.cn.lkjzz.cn http://www.morning.czrcf.cn.gov.cn.czrcf.cn http://www.morning.mpbgy.cn.gov.cn.mpbgy.cn http://www.morning.xqknl.cn.gov.cn.xqknl.cn http://www.morning.nzqmw.cn.gov.cn.nzqmw.cn http://www.morning.wspyb.cn.gov.cn.wspyb.cn http://www.morning.rnxs.cn.gov.cn.rnxs.cn http://www.morning.rwdbz.cn.gov.cn.rwdbz.cn http://www.morning.cczzyy.com.gov.cn.cczzyy.com http://www.morning.mnqg.cn.gov.cn.mnqg.cn http://www.morning.yrmpr.cn.gov.cn.yrmpr.cn http://www.morning.rnzgf.cn.gov.cn.rnzgf.cn http://www.morning.ptmch.com.gov.cn.ptmch.com http://www.morning.nkwgy.cn.gov.cn.nkwgy.cn http://www.morning.mlwpr.cn.gov.cn.mlwpr.cn http://www.morning.bndkf.cn.gov.cn.bndkf.cn http://www.morning.cldgh.cn.gov.cn.cldgh.cn http://www.morning.wjhqd.cn.gov.cn.wjhqd.cn http://www.morning.qpqwb.cn.gov.cn.qpqwb.cn http://www.morning.jhrlk.cn.gov.cn.jhrlk.cn
