《表1 各类仪表数据分布及参数》
实验在Ubuntu 16.04.5+Ge Force GTX1080TI x2+pytorch 1.3.0平台上运行。训练时,将训练数据按9:1的比例划分成训练集和验证集,每类仪表的数据分布如表1所示。对于训练集数据,本文主要进行空间上的数据增强。具体地,给定一张训练图像,进行仪表区域上的正方形随机crop (截取),然后将crop图像resize (调整大小)为256×256。特别地,对于C型仪表,进行宽高比为2:1的长方形随机crop,并resize为256×512,相应的仪表指针方向回归模型最后一层全连接层的输入维度为1 024×2×4。训练时,采用Adam优化器,学习率取1×10-3,batchsize取50,在训练集上训练120个epoch,每个epoch完成时在验证集上测试,最后选取所有epoch中验证精度最高的模型作为最终模型进行测试。每类仪表训练一个读数识别模型。
图表编号 | XD00197606200 严禁用于非法目的 |
---|---|
绘制时间 | 2020.12.10 |
作者 | 彭昆福、王子磊、王磊、顾杨 |
绘制单位 | 中国科学技术大学信息科学技术学院、中国科学技术大学信息科学技术学院、国网安徽省电力有限公司检修分公司、国网安徽省电力有限公司检修分公司 |
更多格式 | 高清、无水印(增值服务) |