From 8f7507eb134ad3df8336931e388646647cc00727 Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@126.com>
Date: Sat, 3 Aug 2019 23:28:10 -0700
Subject: [PATCH] manually add type promotion (#1114)
---
mmdet/core/anchor/anchor_target.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/mmdet/core/anchor/anchor_target.py b/mmdet/core/anchor/anchor_target.py
index 91238e6..bba372f 100644
--- a/mmdet/core/anchor/anchor_target.py
+++ b/mmdet/core/anchor/anchor_target.py
@@ -164,10 +164,10 @@ def anchor_inside_flags(flat_anchors, valid_flags, img_shape,
img_h, img_w = img_shape[:2]
if allowed_border >= 0:
inside_flags = valid_flags & \
- (flat_anchors[:, 0] >= -allowed_border) & \
- (flat_anchors[:, 1] >= -allowed_border) & \
- (flat_anchors[:, 2] < img_w + allowed_border) & \
- (flat_anchors[:, 3] < img_h + allowed_border)
+ (flat_anchors[:, 0] >= -allowed_border).type(torch.uint8) & \
+ (flat_anchors[:, 1] >= -allowed_border).type(torch.uint8) & \
+ (flat_anchors[:, 2] < img_w + allowed_border).type(torch.uint8) & \
+ (flat_anchors[:, 3] < img_h + allowed_border).type(torch.uint8)
else:
inside_flags = valid_flags
return inside_flags
--
GitLab