Skip to content

GitLab

Unverified Commit e6e23152 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

fix inplace add in roi extractor (#1160)

parent 31bf963d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet/models/roi_extractors/single_level.py
View file @ e6e23152
...@@ -103,5 +103,5 @@ class SingleRoIExtractor(nn.Module): ...@@ -103,5 +103,5 @@ class SingleRoIExtractor(nn.Module):
if inds.any(): if inds.any():
rois_ = rois[inds, :] rois_ = rois[inds, :]
roi_feats_t = self.roi_layers[i](feats[i], rois_) 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 return roi_feats
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment