一种360度传送带快速缺陷检测方法及装置

专利主权项内容

1.一种360度传送带快速缺陷检测方法，其特征在于，包括如下步骤：S1、在传送带上方可覆盖传送带上物品四周360度的不同方位处设置相机，当传送带上的物品经过相机时，触发相机拍照，获取相机图片，一个相机一张图片，将所有图片放入一个batch中，并对batch进行归一化操作；S2、将归一化后的图片输入轻量级网络yolov5nano中，对图片中的物品进行位置定位，根据定位结果对原图片进行裁剪，去除图片中其它无关信息，再对图片进行放大操作；S3、将经S2步骤得到的图片重新组成新的batch输入到yolov5s检测网络中，对图片中的物品进行缺陷检测；S4、获取缺陷检测结果，剔除机构根据剔除信号剔除有缺陷物品，记录物品的总数量、缺陷物品数量、缺陷类型和对应缺陷类型的数量；开启主线程，检测相机的数量，为每台相机创建一个线程用于获取图片数据流，并为每个线程编号；申请CPU内存和GPU显存用于存放需要检测的图片，分别为CPU输入内存、GPU输入显存、CPU输出内存、GPU输出显存，其大小分别为一个batch中所有图片输入所占位数、模型输出所占位数，并为每个线程的输入图片设置存放位置；当物品经过相机触发拍照时，调用每个相机的线程，根据线程的编号将该线程的图片数据放入CPU输入内存对应的位置，当内存未填满时将阻塞线程，直至CPU输入内存填满一个batch中设定的所有图片；将CPU输入内存中的数据传到GPU输入显存中，经过轻量级网络yolov5nano进行推理后，将最终输出的推理数据从GPU输出显存中传到CPU输出内存中，解除线程阻塞，根据线程编号将获取的数据进行图像后处理。