联系人:
所在地:
本发明涉及基于YOLOv3剪枝网络和嵌入式开发板的实时药盒检测方法。包括如下步骤:步骤一:设计YOLOv3主干网络及损失函数;步骤二:通过人工拍摄的方式对各个品牌的药盒进行图像数据采集;步骤三:制作药盒数据集并进行训练;步骤四:通过基于BN层缩放因子γ的剪枝方法对YOLOv3进行模型压缩和加速推算;步骤五:将YOLOv3压缩模型部署到Nano嵌入式系统上并使用TensorRT进行模型推理加速;步骤六:使用CSI摄像头在Nano上进行实时药盒检测。本发明用于部署在NVIDIA Jetson Nano嵌入式开发板上进行实时药盒检测,在保证检测精度的同时,确保了检测的实时性以及模型运算的高效率。