From f20e7fb31abe444d9204ce92665af43265f284d9 Mon Sep 17 00:00:00 2001 From: HuguesRoy <149707970+HuguesRoy@users.noreply.github.com> Date: Tue, 4 Jun 2024 13:53:33 +0200 Subject: [PATCH] Update quality_check.py (#609) * Update quality_check.py --- clinicadl/quality_check/t1_linear/quality_check.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/clinicadl/quality_check/t1_linear/quality_check.py b/clinicadl/quality_check/t1_linear/quality_check.py index c684858ea..86d85366d 100755 --- a/clinicadl/quality_check/t1_linear/quality_check.py +++ b/clinicadl/quality_check/t1_linear/quality_check.py @@ -141,7 +141,10 @@ def quality_check( qc_df = pd.DataFrame(columns=columns) qc_df["pass"] = qc_df["pass"].astype(bool) softmax = torch.nn.Softmax(dim=1) - logger.info(f"Quality check will be performed over {len(dataloader)} images.") + + logger.info( + f"Quality check will be performed over {len(dataloader.dataset)} images." + ) for data in dataloader: logger.debug(f"Processing subject {data['participant_id']}.")