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']}.")