refactor: vedem ocr get texts

This commit is contained in:
debugtalk
2022-10-12 20:25:02 +08:00
parent f6623c0869
commit daa62486c5
6 changed files with 28 additions and 59 deletions
+5 -9
View File
@@ -41,22 +41,18 @@ func (dExt *DriverExt) GetTextXY(ocrText string, index ...int) (point PointF, er
return point, nil
}
func (dExt *DriverExt) GetTextXYs(ocrText []string) (points map[string]PointF, err error) {
func (dExt *DriverExt) GetTextXYs(ocrText []string) (points []PointF, err error) {
ps, err := dExt.FindTextsByOCR(ocrText)
if err != nil {
return map[string]PointF{}, err
return nil, err
}
points = map[string]PointF{}
for text, point := range ps {
if len(point) == 0 {
points[text] = PointF{}
continue
}
points[text] = PointF{
for _, point := range ps {
pointF := PointF{
X: point[0] + point[2]*0.5,
Y: point[1] + point[3]*0.5,
}
points = append(points, pointF)
}
return points, nil