diff --git a/src/distributed_hierarchical_attentive/main/tester/multi_lane_test.py b/src/distributed_hierarchical_attentive/main/tester/multi_lane_test.py index f0fbc49a75..4c42ea8a28 100644 --- a/src/distributed_hierarchical_attentive/main/tester/multi_lane_test.py +++ b/src/distributed_hierarchical_attentive/main/tester/multi_lane_test.py @@ -116,7 +116,7 @@ def compute_transform_matrix(self): # 鸟瞰图的四个角点 output_points = np.array([ [0, self.output_size[0]], # 左下 - [self.output_size[1], self.output_size[0]], # 右下 + [*self.output_size[::-1]], # 右下 (w, h) [self.output_size[1], 0], # 右上 [0, 0] # 左上 ], dtype=np.float32)