def is_route_advisor_visible(self, scr, threshold=2):
ra = self.get_route_advisor_image(scr)
#misc.imshow(ra)
#print("ra_shape", ra.shape)
#assert ra.shape == (9, 3)
#ra1d = np.average(ra, axis=2)
ra_rows = np.average(ra, axis=1)
#print("ra_rows.shape", ra_rows.shape)
#print("ra_rows", ra_rows)
expected = np.array([[ 25.33766234, 22.92207792, 21.94805195],
[ 31.79220779, 29.50649351, 28.58441558],
[ 70.32467532, 68.96103896, 68.32467532],
[ 63.51948052, 61.97402597, 61.2987013 ],
[ 66.20779221, 64.72727273, 64.14285714],
[ 64.12987013, 62.51948052, 62.01298701],
[ 60.61038961, 58.94805195, 58.20779221],
[ 65.31168831, 63.74025974, 63.12987013],
[ 18.18181818, 15.66233766, 14.51948052]], dtype=np.float32)
#print("expected", ra_rows)
#print("diff", ra_rows - expected)
# evade brightness differences
observed_normalized = ra_rows - np.average(ra_rows)
expected_normalized = expected - np.average(expected)
#print("observed_normalized", observed_normalized)
#print("expected_normalized", expected_normalized)
dist = np.abs(observed_normalized - expected_normalized)
dist_avg = np.average(dist)
#print("dist", dist)
#print("dist_avg", dist_avg)
return dist_avg < threshold
# quite close scores even for some non-paused images
评论列表
文章目录