Update

open-mmlab · Aug 29, 2022 · f829736 · f829736
1 parent 3de1d0c
commit f829736
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmdet/models/detectors/base.py b/mmdet/models/detectors/base.py
@@ -152,4 +152,5 @@ def add_pred_to_datasample(self, data_samples: SampleList,
         """
         for data_sample, pred_instances in zip(data_samples, results_list):
             data_sample.pred_instances = pred_instances
+        samplelist_boxlist2tensor(data_samples)
         return data_samples