【实战指南】CCPD数据集车牌检测框坐标解析与YOLO格式转换技巧
1. CCPD数据集车牌检测实战入门第一次接触CCPD数据集时我被它独特的文件名编码方式吸引了。这个数据集包含了超过30万张真实场景下的车牌图片每张图片的文件名都像是一个密码本藏着车牌位置的关键信息。比如025-95449_186473-74525_186473-86464_154441-0_0_5_32_30_31_30_30-37-38.jpg这样的文件名新手看到可能会头皮发麻但其实掌握了规律后你会发现这比手动标注高效多了。CCPD数据集最大的特点就是把标注信息直接编码在文件名里省去了单独的标注文件。文件名中包含车牌检测框的左上角和右下角坐标格式固定为x1y1_x2y2。比如95449_186473就表示检测框左上角坐标(95,449)和右下角坐标(186,473)。这种设计让数据预处理变得非常高效但也需要我们掌握正确的解析方法。2. 文件名结构深度解析2.1 坐标信息提取技巧CCPD的文件名结构看似复杂其实可以分解为几个明确的部分。以025-95449_186473-...jpg为例开头的025是图片序号第一个-后的95449_186473就是我们要的车牌区域坐标后续部分包含车牌字符等信息对检测任务可以先忽略提取坐标的关键是定位两个坐标对之间的位置。我推荐使用正则表达式来精准捕获这个模式import re filename 025-95449_186473-74525_186473.jpg match re.search(r(\d\d_\d\d), filename) if match: coords match.group(1) # 得到95449_1864732.2 异常情况处理实际处理时会遇到各种意外情况文件名格式不符预期坐标值超出图片范围特殊字符导致解析失败我建议添加健壮性检查def validate_coords(x1, y1, x2, y2, img_width, img_height): # 检查坐标是否在合理范围内 x1, y1 max(0, x1), max(0, y1) x2, y2 min(img_width-1, x2), min(img_height-1, y2) # 检查是否是有效矩形 if x2 x1 or y2 y1: raise ValueError(Invalid bounding box coordinates) return x1, y1, x2, y23. YOLO格式转换核心算法3.1 坐标归一化原理YOLO需要的不是绝对坐标而是归一化后的相对坐标。具体要转换四个值中心点x坐标 (x1x2)/2/图像宽度中心点y坐标 (y1y2)/2/图像高度框宽度 (x2-x1)/图像宽度框高度 (y2-y1)/图像高度这个转换过程看似简单但小数点处理很关键。我建议保留6位小数def convert_to_yolo(x1, y1, x2, y2, img_w, img_h): x_center (x1 x2) / 2 / img_w y_center (y1 y2) / 2 / img_h width (x2 - x1) / img_w height (y2 - y1) / img_h return round(x_center, 6), round(y_center, 6), round(width, 6), round(height, 6)3.2 批量处理实战代码完整的处理流程应该包括遍历图片目录解析每张图片的尺寸提取文件名中的坐标转换为YOLO格式保存到对应txt文件这是我优化后的完整代码import os import cv2 import re def process_ccpd_to_yolo(src_dir, dest_dir): images_dir os.path.join(dest_dir, images) labels_dir os.path.join(dest_dir, labels) os.makedirs(images_dir, exist_okTrue) os.makedirs(labels_dir, exist_okTrue) for filename in os.listdir(src_dir): if not filename.lower().endswith((.jpg, .png)): continue img_path os.path.join(src_dir, filename) img cv2.imread(img_path) if img is None: continue h, w img.shape[:2] # 提取坐标 match re.search(r(\d\d_\d\d), filename) if not match: continue x1, y1, x2, y2 map(int, match.group(1).replace(, _).split(_)) x1, y1, x2, y2 validate_coords(x1, y1, x2, y2, w, h) # 转换YOLO格式 x, y, bw, bh convert_to_yolo(x1, y1, x2, y2, w, h) # 保存标签文件 base_name os.path.splitext(filename)[0] label_file os.path.join(labels_dir, f{base_name}.txt) with open(label_file, w) as f: f.write(f0 {x} {y} {bw} {bh}) # 0是类别ID # 复制图片到新目录 new_img_path os.path.join(images_dir, filename) cv2.imwrite(new_img_path, img)4. 与YOLO训练流程衔接4.1 数据集组织规范YOLOv5/v8等现代框架对数据集目录结构有明确要求。建议按如下方式组织ccpd_yolo/ ├── images/ │ ├── train/ │ ├── val/ │ └── test/ └── labels/ ├── train/ ├── val/ └── test/我习惯使用80-10-10的比例划分数据集。可以使用这个函数自动划分import shutil import random from sklearn.model_selection import train_test_split def split_dataset(src_images, src_labels, dest_root, test_size0.2, val_size0.1): all_files [f for f in os.listdir(src_images) if f.endswith((.jpg,.png))] train_val, test train_test_split(all_files, test_sizetest_size) train, val train_test_split(train_val, test_sizeval_size/(1-test_size)) for split, files in [(train, train), (val, val), (test, test)]: img_dir os.path.join(dest_root, images, split) label_dir os.path.join(dest_root, labels, split) os.makedirs(img_dir, exist_okTrue) os.makedirs(label_dir, exist_okTrue) for f in files: # 移动图片 shutil.copy(os.path.join(src_images, f), img_dir) # 移动标签 label_file os.path.splitext(f)[0] .txt shutil.copy(os.path.join(src_labels, label_file), label_dir)4.2 YOLO配置文件调整转换完成后需要在YOLO配置文件中正确指定类别数量(nc)类别名称(names)训练/验证集路径典型的data.yaml配置示例# CCPD数据集配置 train: ../ccpd_yolo/images/train val: ../ccpd_yolo/images/val test: ../ccpd_yolo/images/test nc: 1 # 只有车牌一个类别 names: [license_plate] # 类别名称5. 常见问题与性能优化5.1 处理效率提升技巧当处理数十万张图片时原始方法可能很慢。我总结了几个加速技巧多进程处理使用Python的multiprocessingfrom multiprocessing import Pool def process_file(args): filename, src_dir, dest_dir args # 处理单个文件的代码... if __name__ __main__: files [(f, src_dir, dest_dir) for f in os.listdir(src_dir)] with Pool(processes8) as pool: pool.map(process_file, files)使用更快的图片库用TurboJpeg替代OpenCVfrom turbojpeg import TurboJPEG jpeg TurboJPEG() with open(image.jpg, rb) as f: img jpeg.decode(f.read())批量读取图片尺寸对于仅需要尺寸的情况def get_image_size(filepath): with Image.open(filepath) as img: return img.size # (width, height)5.2 特殊场景处理真实项目中会遇到各种特殊情况倾斜车牌CCPD中部分车牌有旋转检测框可能不是标准矩形多车牌同框需要处理一个图片中出现多个车牌的情况夜间低质量图片需要额外的预处理对于倾斜车牌可以考虑改用旋转矩形表示法# 使用四点坐标表示旋转矩形 def convert_rotated_bbox(points, img_w, img_h): # points是四个角点坐标 # 计算最小外接矩形 rect cv2.minAreaRect(np.array(points)) # 转换为YOLO旋转格式 (x_center, y_center, width, height, angle) angle rect[2] # 旋转角度 return (*convert_to_yolo(*cv2.boundingRect(np.array(points)), img_w, img_h), angle)处理CCPD数据集最关键的还是理解它的文件名编码规则。我在实际项目中发现约5%的图片可能需要手动修正坐标建议在转换完成后随机抽查一批样本用这个可视化代码检查def visualize_bbox(img_path, label_path): img cv2.imread(img_path) h, w img.shape[:2] with open(label_path) as f: cls, x, y, bw, bh map(float, f.read().split()) # 转换回绝对坐标 x1 int((x - bw/2) * w) y1 int((y - bh/2) * h) x2 int((x bw/2) * w) y2 int((y bh/2) * h) cv2.rectangle(img, (x1,y1), (x2,y2), (0,255,0), 2) cv2.imshow(check, img) cv2.waitKey(0)
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2441706.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!