Skip to content

Commit 2d72cca

Browse files
committed
Update det_tta.py
1 parent c53486b commit 2d72cca

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mmdet/models/test_time_augs/det_tta.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,9 @@ def _merge_single_sample(
112112
aug_scores = []
113113
aug_labels = []
114114
img_metas = []
115+
# TODO: support instance segmentation TTA
116+
assert data_samples[0].pred_instances.get('masks', None) is None, \
117+
'TTA of instance segmentation does not support now.'
115118
for data_sample in data_samples:
116119
aug_bboxes.append(data_sample.pred_instances.bboxes)
117120
aug_scores.append(data_sample.pred_instances.scores)

0 commit comments

Comments
 (0)