diff --git a/week18/dataset.py b/week18/dataset.py
index ef32ffdf8f48a0785f598df70e8b2fabba9bf016..031b7ee02ac2c7e9677fb124d4d628225d7e3eac 100644
--- a/week18/dataset.py
+++ b/week18/dataset.py
@@ -370,11 +370,11 @@ def get_score_geo(img, vertices, labels, scale, length):
         # 计算d，负数表示在文本框外侧，置零
         d1 = rotated_y - y_min
         d1[d1<0] = 0
-        d2 = y_max - rotated_y
+        d2 = x_max - rotated_x
         d2[d2<0] = 0
-        d3 = rotated_x - x_min
+        d3 = y_max - rotated_y
         d3[d3<0] = 0
-        d4 = x_max - rotated_x
+        d4 = rotated_x - x_min
         d4[d4<0] = 0
         # 每隔4个像素采样(index_y, index_x)，乘以文本框缩小0.3后的掩码
         geo_map[:,:,0] += d1[index_y, index_x] * temp_mask