diff --git a/mmedit/evaluation/metrics/ms_ssim.py b/mmedit/evaluation/metrics/ms_ssim.py index b14bfd103e..06cf6e517a 100644 --- a/mmedit/evaluation/metrics/ms_ssim.py +++ b/mmedit/evaluation/metrics/ms_ssim.py @@ -299,12 +299,6 @@ def process(self, data_batch: dict, data_samples: Sequence[dict]) -> None: minibatch = torch.stack(fake_imgs, dim=0) assert minibatch.shape[0] % 2 == 0, 'batch size must be divided by 2.' - # minibatch = ((minibatch + 1) / 2) - # minibatch = minibatch.clamp_(0, 1) - # half1 = minibatch[0::2].cpu().data.numpy().transpose((0, 2, 3, 1)) - # half1 = (half1 * 255).astype('uint8') - # half2 = minibatch[1::2].cpu().data.numpy().transpose((0, 2, 3, 1)) - # half2 = (half2 * 255).astype('uint8') half1 = minibatch[0::2].cpu().data.numpy().transpose((0, 2, 3, 1)) half2 = minibatch[1::2].cpu().data.numpy().transpose((0, 2, 3, 1))