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

基建选人增加日志debug输出
All checks were successful
ci/woodpecker/push/check_format Pipeline was successful
Details

Browse source
This commit is contained in:
EightyDollars 2024-12-05 20:05:31 +08:00
parent dc9f06512a
commit de4a699098
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
mower/solvers/infra/base_choose.py
View file

@ -100,7 +100,9 @@ class RIIC_ChooseSolver(SceneGraphSolver, BaseMixin):
return return
if agent_name in agents_dict.keys(): if agent_name in agents_dict.keys():
if self.is_choosed(agents_dict[agent_name]): is_choosed = self.is_choosed(agents_dict[agent_name])
logger.debug(f"{agent_name} 是否被选上：{is_choosed}")
if is_choosed:
logger.info(f"{agent_name}选择完成") logger.info(f"{agent_name}选择完成")
self.choosed.append(agent_name) self.choosed.append(agent_name)
self.agent = None self.agent = None
@ -154,22 +156,23 @@ class RIIC_ChooseSolver(SceneGraphSolver, BaseMixin):
raise RuntimeError(f"缺少干员 {self.agent[0]}") raise RuntimeError(f"缺少干员 {self.agent[0]}")
def is_choosed(self, scope) -> bool: def is_choosed(self, scope) -> bool:
if ( up_pt = cropimg(
cropimg( config.recog.img,
config.recog.img, [
[ [scope[0][0] + 0, scope[0][1] - 128],
[scope[0][0] + 0, scope[0][1] - 128], [scope[0][0] + 1, scope[0][1] - 127],
[scope[0][0] + 1, scope[0][1] - 127], ],
], )[0][0][0]
)[0][0][0]
< 50 down_pt = cropimg(
) and cropimg(
config.recog.img, config.recog.img,
[ [
[scope[0][0] + 0, scope[1][1] + 212], [scope[0][0] + 0, scope[1][1] + 212],
[scope[0][0] + 1, scope[1][1] + 213], [scope[0][0] + 1, scope[1][1] + 213],
], ],
)[0][0][0] < 50: )[0][0][0]
logger.debug(f"is_choosed: {up_pt}{down_pt}")
if (up_pt < 50) and down_pt < 50:
return True return True
return False return False