【工业机器视觉】基于深度学习的水表盘读数识别(3-数据标注与转换)-CSDN博客
读数修正
上一篇我们已经正确对图片进行了预测,并且可视化了预测结果,接下来我们要对预测结果数据进行解析、处理、修正,得到最终读数。这里不再介绍上位机开发,本文是基于Pyside6开发的上位机。
模型加载
分别加载3个训练好的模型:
- self.modelPointer = YOLO(model_path + weight_pointer)
- self.modelSeg = YOLO(model_path + weight_seg)
- self.modelDigit = YOLO(model_path + weight_digit)
modelPointer:指针目标检测
modelSeg:梅花针分割
modelDigit:字轮数字目标检测
评论记录:
回复评论: