基于改进YOLOv5的电子元件表面缺陷检测算法

曾耀,高法钦

Surface defect detection algorithm of electronic components based on improved YOLOv5

Yao ZENG,Fa-qin GAO

表 1 YOLOv5模型训练实验环境配置表

Tab.1 Configuration of YOLOv5 model training experimental environment

配置对象 配置内容 操作系统 Windows 10 处理器 NVIDIA GeForce GTX 1050 开发环境 PyCharm 深度学习框架 Pytorch 1.7.1 CUDA Cuda 10.1 CUDNN Cudnn 7.6.5 采样相机 海康工业相机MV-CE050-31GM