Update quality_check.py (#609)

* Update quality_check.py
aramis-lab · Jun 4, 2024 · f20e7fb · f20e7fb
1 parent d54d59c
commit f20e7fb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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']}.")