llzzyy233 commited on
Commit
c0f69f8
·
verified ·
1 Parent(s): 371b59e

更新 readme

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