Update GANDLF/compute/inference_loop.py

Co-authored-by: Sarthak Pati <[email protected]>
mlcommons · Oct 31, 2024 · adcd561 · adcd561
1 parent bf7abfc
commit adcd561
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/GANDLF/compute/inference_loop.py b/GANDLF/compute/inference_loop.py
@@ -344,6 +344,7 @@ def inference_loop(
                         )
                         cv2.imwrite(file_to_write, heatmaps[key])
 
+                        # this is needed because openslide returns an RGBA image
                         os_image_array = np.asarray(
                             os_image.read_region(
                                 (0, 0),