首页 > 解决方案 > 如何将解码的数据矩阵写入数据帧

问题描述

我正在尝试从图像中读取所有数据矩阵并写入数据帧。我可以通过 pylibdmtx 打印条形码编号和位置,但我不知道如何存储在数据框中

image = cv2.imread('IMG-4951.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
plt.imshow(gray)
ret,thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
msg = pylibdmtx.decode(thresh)
print(msg)


Output:
[Decoded(data=b'01086995460155972150046789702417240229109LB02199', rect=Rect(left=984, top=1172, width=290, height=287)), Decoded(data=b'01086995460155972154702360250417240229109LB02199', rect=Rect(left=899, top=2242, width=279, height=272))]

在这种情况下,'msg' 变量存储为具有 2 个元素的列表,当我尝试转换 pandas Dataframe 时,'data' 列是空白的,但 'rect' 列像上面一样正确。(Rect(left=984, top=1172, width=290, height=287))

数据框如下所示;

data      rect
          Rect(left=984, top=1172, width=290, height=287)
          Rect(left=899, top=2242, width=279, height=272)

如何填写数据列或您建议的任何其他方法?

我的第二个问题是,这个库似乎很慢,有什么建议可以让它更快吗?

提前致谢,

标签: pythonpandasdataframedatamatrix

解决方案


这可能会有所帮助

我使用了这个测试图像

在此处输入图像描述

导入包

import cv2
import matplotlib.pyplot as plt
from pylibdmtx.pylibdmtx import decode
import pandas as pd 

使用您在上面提供的代码

image = cv2.imread('datamatrix.png')
gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
plt.imshow(gray)
ret,thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
msg = decode(thresh)
print(msg)

我得到这个输出,

[Decoded(data=b'Stegosaurus', rect=Rect(left=5, top=6, width=96, height=95)), Decoded(data=b'Plesiosaurus', rect=Rect(left=298, top=6, width=95, height=95))]

在此处输入图像描述

要添加到数据框,我首先填充一个列表来表示我希望数据框成为的形状

ls_msg = []
for item in msg:
    ls_msg.append([item[0], item[1][0], item[1][1], item[1][2]])

将列表制作成数据框

df_msg = pd.DataFrame(ls_msg)

添加列名

df_msg.columns = ['data', 'left','top','width']

这会产生一个看起来像这样的数据框

在此处输入图像描述

在安装和播放令人着迷之前从未使用过此软件包!


推荐阅读