mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Merge branch 'linfangjian/fix_pred_new' into 'refactor_dev'

Browse Source 
[Fix] Fix pred

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!43
This commit is contained in:
zhengmiao 2022-06-14 02:53:46 +00:00
commit ec52a27299
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/metrics/iou_metric.py
View File

@ -63,7 +63,7 @@ class IoUMetric(BaseMetric):
reduce_zero_label = self.dataset_meta['reduce_zero_label']
for data, pred in zip(data_batch, predictions):
label = data['data_sample']['gt_sem_seg']['data'][0].cpu().numpy()
pred_label = pred['pred_sem_seg']['data'][0]
pred_label = pred['pred_sem_seg']['data'][0].cpu().numpy()
self.results.append(
self.intersect_and_union(pred_label, label, num_classes,
self.ignore_index, label_map,

3
mmseg/models/segmentors/encoder_decoder.py
View File

@ -256,8 +256,7 @@ class EncoderDecoder(BaseSegmentor):
results_dict = dict()
seg_logit = self.inference(batch_inputs, batch_img_metas, rescale)
results_dict['seg_logits'] = seg_logit
seg_pred = seg_logit.argmax(dim=1)
seg_pred = seg_pred.cpu().numpy()
seg_pred = seg_logit.argmax(dim=1, keepdim=True)
results_dict['pred_sem_seg'] = seg_pred
results_list = self.postprocess_result(results_dict)
return results_list

1
tests/test_metrics/test_iou_metric.py
View File

@ -65,6 +65,7 @@ class TestIoUMetric(TestCase):
results_dict['seg_logits'] = seg_logit
seg_pred = np.random.randint(
0, num_classes, (batch_size, h, w), dtype=np.uint8)
seg_pred = torch.LongTensor(seg_pred)
results_dict['pred_sem_seg'] = seg_pred
batch_datasampes = [