Skip to content

Commit

Permalink
Pylint failing
Browse files Browse the repository at this point in the history
  • Loading branch information
georgeyiasemis committed Jun 24, 2024
1 parent 341ceb7 commit e50b77b
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions direct/functionals/challenges.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,9 @@
""" Direct metrics for the FastMRI and Calgary-Campinas challenges."""

import numpy as np
import skimage.metrics
import torch

from skimage.metrics import structural_similarity, peak_signal_noise_ratio

__all__ = (
"fastmri_ssim",
"fastmri_psnr",
Expand All @@ -28,7 +27,7 @@ def fastmri_ssim(gt, target):

gt = _to_numpy(gt)[:, 0, ...]
target = _to_numpy(target)[:, 0, ...]
out = structural_similarity(
out = skimage.metrics.structural_similarity(
gt.transpose(1, 2, 0),
target.transpose(1, 2, 0),
channel_axis=-1,
Expand All @@ -41,9 +40,8 @@ def fastmri_psnr(gt, pred):
"""Compute Peak Signal to Noise Ratio metric (PSNR) compatible with the FastMRI challenge."""
gt = _to_numpy(gt)[:, 0, ...]
pred = _to_numpy(pred)[:, 0, ...]
from skimage.metrics import peak_signal_noise_ratio as psnr

out = psnr(image_true=gt, image_test=pred, data_range=gt.max())
out = skimage.metrics.peak_signal_noise_ratio(image_true=gt, image_test=pred, data_range=gt.max())
return torch.from_numpy(np.array(out)).float()


Expand Down Expand Up @@ -73,11 +71,11 @@ def _calgary_campinas_metric(gt, pred, metric_func):


def calgary_campinas_ssim(gt, pred):
return _calgary_campinas_metric(gt, pred, structural_similarity)
return _calgary_campinas_metric(gt, pred, skimage.metrics.structural_similarity)


def calgary_campinas_psnr(gt, pred):
return _calgary_campinas_metric(gt, pred, peak_signal_noise_ratio)
return _calgary_campinas_metric(gt, pred, skimage.metrics.peak_signal_noise_ratio)


def calgary_campinas_vif(gt, pred):
Expand Down

0 comments on commit e50b77b

Please sign in to comment.