From 25059c278b08f5995258dfb698a45eae0fb0bb9c Mon Sep 17 00:00:00 2001 From: Kyson Lok Date: Fri, 8 May 2020 22:31:50 +0800 Subject: [PATCH] fix: top bottom left right -> top right bottom left --- week18/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/week18/dataset.py b/week18/dataset.py index ef32ffd..031b7ee 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 -- Gitee