HibikiJie / ReadMeter

Read the value of the pointer meter

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

该存储库为识别指针式仪表,见下图

831

输出结果为归一化后的指针与刻度之间的相对位置。

理论上可已处理多指针多表盘的针表,只需要将不同的指针与刻度分配至不同的通道中即可。

**所有代码和模型都在积极开发中,如有更改或删除,恕不另行通知。**使用风险自负。

讲解

  • 语义分割模型,在不同通道上分割出表盘和指针
  • 环形的表盘展开为矩形图像
  • 二维图像转换为一维数组
  • 对刻度数组用均值滤波
  • 定位指针相对刻度的位置
  • 输出相对位置

语义分割模型采用的是U2Net

要求

安装了所有依赖项的Python 3.8或更高版本,包括torch>=1.7。要安装运行:

环境环境

Meter可以在以下任何经过验证的最新环境中运行(预安装所有依赖项,包括CUDA / CUDNNPythonPyTorch):

训练

运行以下命令以在Data文件夹下的数据集上重现结果。在一台GTX2080TI上。使用--batch-size为您的GPU允许的最大容量(为11 GB设备显示的批量大小)。

$ python train.py

推理

运行read_meter.py可以计算data/文件下的val图片,输出的指针数值为归一化后的数值。

权重

链接: https://pan.baidu.com/s/1wTPo1wJXrNyEFSu8RrD8Xw 密码: t0p4

将文件放置于weight下

About

Read the value of the pointer meter


Languages

Language:Python 100.0%