基于YOLOv8的PCB板缺陷檢測(cè)

介紹

基于DeepPCB這個(gè)公開數(shù)據(jù)集，總計(jì)有1500份的模板-缺陷圖像數(shù)據(jù)對(duì)，總計(jì)圖像3000張，對(duì)應(yīng)text格式的1500個(gè)標(biāo)注文本描述文件。包含PCB主要的六個(gè)類別錯(cuò)誤，分別是：

pen
short
mousebite
spur
pin hole
spurious copper

數(shù)據(jù)標(biāo)注的的格式為：x1 y1 x2 y2 type 標(biāo)注數(shù)據(jù)分為兩個(gè)部分1000張作為訓(xùn)練樣本，500張作為做測(cè)試樣本，訓(xùn)練與測(cè)試樣本的缺陷數(shù)據(jù)統(tǒng)計(jì)分布如下：

YOLOv8模型訓(xùn)練

首先基于數(shù)據(jù)集，制作YOLO格式數(shù)據(jù)集，要把標(biāo)注文件從VOC格式轉(zhuǎn)換YOLO格式，然后按照指定的格式制作完成數(shù)據(jù)集。不知道怎么制作的看這里： YOLOv8自定義數(shù)據(jù)集訓(xùn)練到模型部署推理 制作好數(shù)據(jù)集，配置好數(shù)據(jù)集描述文件，

然后直接使用下面的命令行開始訓(xùn)練：

yolo train model=yolov8s.pt data=pcb_dataset.yaml epochs=15 imgsz=640 batch=1

訓(xùn)練完成之后如下：

測(cè)試評(píng)估的結(jié)果如下：