Summary: | Small target detection has become an important research project in image processing and deep learning target detection algorithms in recent years. Different from common small target detection objects, this project uses machine learning methods to detect targets for various components in circuit diagrams. It is different from the traditional identification method of artificially defined target features. It is mainly through training the convolutional neural network model to achieve high-precision feature extraction for complex circuit diagrams containing various components. The YOLOv3 model is You Only Look Once,version 3. It is used as the original framework. A common class of component-connected incomplete circuits is pre-trained on a dataset, and then the model is fine-tuned with complete circuit diagrams. During training, the technology of transfer learning is used to transfer the parameters of the model, which is based on the transfer learning method. In the experimental part, Python is used as the development language, Pytorch is used as the framework of deep learning, and the calculation is performed on the Colab platform in Google. The source of the self-built data set in this dissertation is the image library of Google and Baidu. The quantitative pictures in the data set increase the accuracy of target detection, and the accuracy is gradually improved.The accuracy rate of various classes of recognition can be over 85% or higher.
|