卷積神經網路的印刷電路板元件分類與方位估計;Classification and position estimation for components on printed circuit boards using a convolutional neural network

NCUIR > College of Electrical Engineering & Computer Science > Graduate Institute of Computer Science and Information Engineering > Electronic Thesis & Dissertation > Item 987654321/84018

Please use this identifier to cite or link to this item: http://ir.lib.ncu.edu.tw/handle/987654321/84018

Title:	卷積神經網路的印刷電路板元件分類與方位估計;Classification and position estimation for components on printed circuit boards using a convolutional neural network
Authors:	張晉;Jin, Jhang
Contributors:	資訊工程學系
Keywords:	方位估計;印刷電路板元件;角度;卷積神經網路;YOLOv3
Date:	2020-07-28
Issue Date:	2020-09-02 17:55:51 (UTC+8)
Publisher:	國立中央大學
Abstract:	印刷電路板是當今電子產品的基礎材料，也是電子相關產品不可或缺的重要組件。電路板完成後及插件後都需要個別檢測瑕疵，本研究只考慮插件後的元件檢測項目。在過去的傳統檢測方法中，需要針對特定元件設計出特定合適的演算法，因此檢測速度非常快，但缺點是泛用性很低，對於每種新的元件都得設計新的演算法。近年來深度學習發展快速，且廣泛的應用在各種領域，也漸漸取代了傳統方法。在本研究中，我們使用深度學習的方式來定位及分類印刷電路板上的元件。另外，有些印刷電路板上的元件是有不同方向，為了達到更準確的定位效果，我們發展了旋轉偵測框的定位功能。本研究分為三部份，第一部份是針對標記的修改，印刷電路板上的元件允許有360度的旋轉，我們重新設計訓練影像上的標記；第二部份是針對網路的修改，我們修改YOLOv3的網路架構，以符合新的標記，我們也加入自己設計的角度相關之損失函數，並將GIoU (generalized intersection over union) 及焦點損失函數 (focal loss) 加入原本的損失函數中以加強偵測的效果；第三部份我們使用了資料前處理、標記平滑化、學習率策略及隨機輸入影像尺度的訓練策略，使新的網路能夠對印刷電路板上的元件檢測及估計更加準確。在實驗中分為有角度估計及無角度估計的偵測網路，使用的訓練資料為印刷電路板元件資料集，共有298張影像。無角度估計的偵測網路中，其中267張影像用於訓練，31張影像用於測試；有角度估計的偵測網路中，非0度元件資料過少，298張影像皆用於訓練，無測試影像。偵測與辨識的元件共分26類。在實驗中，無角度估計的偵測網路僅加入兩種損失，且應用四種訓練策略，其中訓練資料的mAP達到99.9%，測試資料的mAP達到89.3%。有角度估計的偵測網路中，亦將訓練資料加入角度的資訊，且在損失函數中加入角度損失，其中訓練資料的mAP達到90.4%。 ;The printed circuit board is the basic and irreplaceable component of electronic products nowadays. There is generally a component detection at the later stage of the component-assembly process for detecting defect. In the past, it was necessary to design a specific suitable algorithm for specific components. This algorithm will be very fast, but the versatility is very low. It means that a new algorithm should be designed for each new component everytimes. Deep learning has developed rapidly, and is widely used in various fields in recent. It has gradually replaced traditional manually algorithms. In this study, we used deep learning to classify and estimate position of components on the printed circuit board. On the printed circuit board, some components have different positions. In order to get more precise position, we have developed a method for these components. There are three parts in our research. First, We redesigned the label of training image for the components on the printed circuit board which have angle. Second, we modified the network structure of YOLOv3 to fit the new label. We also used angle-related, generalized intersection over union and focal loss function to strengthen the effect of detection. Finally, we used some training strategies witch are label smoothing, data pre-processing, learning rate schedule and random shapes training let the network structure get more accuracy for the components on the printed circuit board. In experiment, we used detection networks whether have angle estimation or not. The printed circuit board component dataset be used as training data which have 298 images. In non-angle estimation detection network, there are 267 images used in training and 31 images used in testing. In angle estimation detection network, there are 298 images used in training and no image used in testing. There are 26 classes in our dataset. We modified two loss functions and used four training strategies in non-angle estimation detection networks. The mAP of training data reaches 99.9%, and the mAP of testing data reaches 89.3%. We redesigned the label of images and used angle-related loss function in angle estimation detection networks. The mAP of training data finally reached 90.4%.
Appears in Collections:	[Graduate Institute of Computer Science and Information Engineering] Electronic Thesis & Dissertation

Files in This Item:

File	Description	Size	Format
index.html		0Kb	HTML	127	View/Open

社群 sharing

Loading...