README.md

---
title: PCB DETECT
emoji: 😻
colorFrom: red
colorTo: indigo
sdk: gradio
sdk_version: 4.31.5
app_file: app.py
pinned: false
license: mit
---
# 基于改进YOLOv8算法的PCB缺陷辅助检测系统
传统的PCB缺陷检测主要依赖于人工目检或使用简单的机器视觉系统，但这些方法存在效率低、准确率低、劳动强度大、易受人为因素影响等问题。
单人开发一款基于改进YOLOv8算法的PCB缺陷辅助检测系统，项目提出了一种改进的YOLOv8算法，首先提出一种残差注意力机制，即带有残差连接的CBAM注意力机制ResCBAM，并在YOLOv8模型的头部添加，提高模型提取特征能力的同时，防止模型过拟合。
并使用GhostConv来替换主干中的普通卷积在保证性能的同时使模型更加轻量化。同时引入WIOU损失函数来代替原有的CIOU损失函数，来避免小目标对损失函数影响过大的问题
以及缓解数据集类别不平衡的问题。
通过实验结果表明与原始YOLOv8模型相比，
所提出的改进YOLOv8模型在DeepPCB数据集上的Precision, Recall, F1score, mAP50, mAP50-95分别提高了5.3%，2.3%，3.8%，3.4%，7.3%。显著提升了检测精度与效率，
对保障产品质量、降低生产成本具有重要意义。还将模型部署到了WEB端，用户可以访问网页来上传图片或者拍照进行检测，并可以将结果保存。
项目部署地址：https://huggingface.co/spaces/llzzyy233/PCB-DETECT



项目使用gradio进行部署，依托hugging-face平台 


Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference