HibikiJie / ReadMeter

Read the value of the pointer meter
65 stars 19 forks source link

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

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下