From e6e2315219aac9146f55b37481d0c287f624dd7a Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 11 Aug 2019 10:34:14 +0800
Subject: [PATCH] fix inplace add in roi extractor (#1160)
---
mmdet/models/roi_extractors/single_level.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmdet/models/roi_extractors/single_level.py b/mmdet/models/roi_extractors/single_level.py
index a2ed5c5..6620d1d 100644
--- a/mmdet/models/roi_extractors/single_level.py
+++ b/mmdet/models/roi_extractors/single_level.py
@@ -103,5 +103,5 @@ class SingleRoIExtractor(nn.Module):
if inds.any():
rois_ = rois[inds, :]
roi_feats_t = self.roi_layers[i](feats[i], rois_)
- roi_feats[inds] += roi_feats_t
+ roi_feats[inds] = roi_feats_t
return roi_feats
--
GitLab