mower-ng/mower-ng
4
0
Fork 0
Code Issues 51 Pull requests Wiki Activity

detect_room修复

Browse source
This commit is contained in:
zhbaor 2024-11-24 12:47:54 +08:00
parent e43b90b065
commit 6fed037f09
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
mower/solvers/infra/base_mixin.py
View file

@ -226,12 +226,15 @@ class BaseMixin:
"训练室": 178, "训练室": 178,
"加工站": 32, "加工站": 32,
} }
img = cropimg(config.recog.img, ((568, 18), (957, 95))) scope = ((568, 18), (957, 95))
hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV) img = cropimg(config.recog.img, scope)
gray = cropimg(config.recog.gray, scope)
hsv = cropimg(config.recog.hsv, scope)
colored_room = None colored_room = None
for room, color in color_map.items(): for room, color in color_map.items():
mask = cv2.inRange(hsv, (color - 1, 0, 0), (color + 2, 255, 255)) bin = thres2(gray, 127)
if cv2.countNonZero(mask) > 1000: mask = cv2.inRange(hsv, (color - 2, 0, 0), (color + 2, 255, 255))
if cv2.countNonZero(cv2.bitwise_and(bin, mask)) > 1000:
colored_room = room colored_room = room
break break
if colored_room in ["制造站", "贸易站", "发电站"]: if colored_room in ["制造站", "贸易站", "发电站"]: