Skip to content

GitLab

Commit b932030c authored by yhcao6's avatar yhcao6
Browse files

remove useless assign_results

parent c29ebbc5
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
mmdet/models/detectors/two_stage.py
View file @ b932030c
...@@ -106,7 +106,6 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin, ...@@ -106,7 +106,6 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
bbox_sampler = build_sampler( bbox_sampler = build_sampler(
self.train_cfg.rcnn.sampler, context=self) self.train_cfg.rcnn.sampler, context=self)
num_imgs = img.size(0) num_imgs = img.size(0)
assign_results = []
sampling_results = [] sampling_results = []
for i in range(num_imgs): for i in range(num_imgs):
assign_result = bbox_assigner.assign( assign_result = bbox_assigner.assign(
...@@ -118,7 +117,6 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin, ...@@ -118,7 +117,6 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
gt_bboxes[i], gt_bboxes[i],
gt_labels[i], gt_labels[i],
feats=[lvl_feat[i][None] for lvl_feat in x]) feats=[lvl_feat[i][None] for lvl_feat in x])
assign_results.append(assign_result)
sampling_results.append(sampling_result) sampling_results.append(sampling_result)
# bbox head forward and loss # bbox head forward and loss
......
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