Paddle 印刷电路板(PCB)瑕疵检测

简介: Paddle 印刷电路板(PCB)瑕疵检测

框架介绍


PaddleDetection:飞桨推出的PaddleDetection是端到端目标检测开发套件,旨在帮助开发者更快更好地完成检测模型的训练、精度速度优化到部署全流程。该框架中提供了丰富的数据增强、网络组件、损失函数等模块,集成了模型压缩和跨平台高性能部署能力。目前基于PaddleDetection已经完成落地的项目涉及工业质检、遥感图像检测、无人巡检等多个领域。


印刷电路板(PCB)瑕疵数据集


印刷电路板(PCB)瑕疵数据集:数据下载链接,是一个公共的合成PCB数据集,由北京大学发布,其中包含1386张图像以及6种缺陷(缺失孔,鼠咬伤,开路,短路,杂散,伪铜),用于检测、分类和配准任务。我们选取了其中适用于检测任务的693张图像,随机选择593张图像作为训练集,100张图像作为验证集。


本项目选取yolov3_darknet.yml作为本项目的训练配置文件。该配置文件选取的是YOLOv3_darknet模型,骨干网络为DarkNet,yolo_head为YOLOv3Head。

yolov3_darknet_baseline.yml配置文件修改适配:
    max_iters: 2669
    num_classes: 6
    base_lr: 0.00025
    milestones: [1779, 2372]
yolov3_darknet_baseline.yml的Reader修改适配:
TrainReader:
    image_dir: images
    anno_path: Annotations/train.json
    dataset_dir: PCB_DATASET
EvalReader:
    image_dir: images
    anno_path: Annotations/val.json
    dataset_dir: PCB_DATASET
TestReader:
  anno_path: PCB_DATASET/Annotations/val.json


Anchor重新聚类


python tools/anchor_cluster.py -c ../yolov3_darknet_baseline.yml -n 9 -s 608 -m v2 -i 1000

结果如下:

anchors: [[8, 14], [13, 14], [10, 21],
           [21, 13], [16, 20], [12, 31],
           [30, 17], [22, 27], [36, 36]]


Anchor重新聚类完成后,修改配置文件中YOLOv3Head里的anchors字段,和TrainReader的Gt2YoloTarget中anchors字段。


训练与评估:

python -u tools/train.py -c yolov3_darknet_baseline.yml --eval
python -u tools/eval.py -c yolov3_darknet_baseline.yml -o weights=output/yolov3_darknet_baseline/best_model


最终评估结果如下图所示:


从上图中,可以看出YOLOV3_darknet模型在印刷电路板(PCB)瑕疵数据集数据集上的mAP = 34.9


补充


awesome-DeepLearning:一站式深度学习在线百科,内容涵盖零基础入门深度学习、产业实践深度学习、特色课程;深度学习百问、产业实践(开发中) 等等。从理论到实践,从科研到产业应用,各类学习材料一应俱全,旨在帮助开发者高效地学习和掌握深度学习知识,快速成为AI跨界人才。

目录
相关文章
|
11月前
|
机器学习/深度学习 数据采集 算法
PCB电路板缺陷检测数据集(近千张图片已划分、已标注)| 适用于YOLO系列深度学习检测任务【数据集分享】
在现代电子制造中,印刷电路板(PCB)是几乎所有电子设备的核心组成部分。随着PCB设计复杂度不断增加,人工检测PCB缺陷不仅效率低,而且容易漏检或误判。因此,利用计算机视觉和深度学习技术对PCB缺陷进行自动检测成为行业发展的必然趋势。
PCB电路板缺陷检测数据集(近千张图片已划分、已标注)| 适用于YOLO系列深度学习检测任务【数据集分享】
|
3月前
|
机器学习/深度学习 编解码 人工智能
23类 PCB电子元器件检测数据集(1600张)|YOLO训练数据集 工业质检 小目标检测
本数据集含1600张真实PCB图像,涵盖23类电子元器件,标注精准、格式规范(YOLO标准),专为小目标密集检测优化,适用于YOLOv5/v8等模型训练,助力工业质检、缺陷识别与AI项目开发。
23类 PCB电子元器件检测数据集(1600张)|YOLO训练数据集 工业质检 小目标检测
|
4月前
|
机器学习/深度学习 人工智能 新制造
半导体芯片表面缺陷检测数据集(2500张高质量标注)| YOLO工业视觉数据集 AI质检
本数据集含2500张真实产线采集的半导体芯片图像,YOLO格式标注,涵盖ZF-scratch、scratch、broken、pinbreak四类表面缺陷,标注精度≥99%,适配YOLOv8等模型,专为工业质检、小目标检测与AI质检系统落地设计。
|
机器学习/深度学习 数据采集 数据可视化
基于YOLOv8的PCB缺陷检测识别项目|完整源码数据集+PyQt5界面+完整训练流程+开箱即用!
本项目基于YOLOv8实现PCB缺陷检测,提供一站式解决方案。包含完整训练代码、标注数据集、预训练权重及PyQt5图形界面,支持图片、文件夹、视频和摄像头四种检测模式。项目开箱即用,适合科研、工业与毕业设计。核心功能涵盖模型训练、推理部署、结果保存等,检测类型包括缺孔、鼠咬缺口、开路、短路、飞线和杂铜。项目具备高性能检测、友好界面、灵活扩展及多输入源支持等优势,未来可优化模型轻量化、多尺度检测及报告生成等功能。
基于YOLOv8的PCB缺陷检测识别项目|完整源码数据集+PyQt5界面+完整训练流程+开箱即用!
|
数据采集 存储 机器学习/深度学习
最新AI大模型数据集解决方案:分享两种AI高质量代码数据集生产方案
本文分享了两种构建高质量AI代码数据集的解决方案。第一种是传统方式,结合动态住宅代理与手动处理,通过分页读取和数据清洗生成结构化数据;第二种是利用Web Scraper API工具,实现自定义配置、自动化抓取及云端存储。两种方法各具优势,适合不同需求和技术水平的团队。同时,文章还提供了专属优惠福利,助力提升数据采集效率,为AI大模型训练提供支持。
624 5
最新AI大模型数据集解决方案:分享两种AI高质量代码数据集生产方案
EMQ
|
存储 JSON 数据库
MQTTX 1.10.0 发布:CLI高级文件管理与配置
在本次更新中,CLI 版本在文件管理和配置功能方面进行了显著增强。主要更新包括:支持从文件中读取和写入消息、高级配置选项、文本输出模式、以及改进的日志记录。此外,桌面版本现在支持数据库重建,以防止文件损坏引起的问题,并且能更好地处理大数据的展示。这些更新希望为所有 MQTTX 用户提供更加强大和用户友好的体验。
EMQ
1217 251
MQTTX 1.10.0 发布:CLI高级文件管理与配置
|
编解码 人工智能 测试技术
CogView4:智谱开源中文文生图新标杆,中文海报+任意分辨率一键生成
CogView4 是智谱推出的开源文生图模型,支持中英双语输入和任意分辨率图像生成,特别优化了中文文字生成能力,适合广告、创意设计等场景。
917 1
CogView4:智谱开源中文文生图新标杆,中文海报+任意分辨率一键生成
|
人工智能 Java 程序员
一文彻底搞明白电路图和原理图
本文介绍电路图的基本构成、作用及类型,帮助理解电路原理和实际应用。元件符号、连线、节点和注释说明是电路图的核心要素,通过原理图、方框图、装配图和印板图,可以清晰展示电路结构和工作原理。君志所向,一往无前!
12380 0
|
监控 安全 大数据
关于微信支付相关安全性问题
关于微信支付相关安全性问题
关于微信支付相关安全性问题
|
存储 数据中心 云计算

热门文章

最新文章