Added a todo comment

Jonathunky · Nov 3, 2023 · 57089af · 57089af
1 parent 44db797
commit 57089af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_loader.py b/data_loader.py
@@ -271,7 +271,7 @@ def __getitem__(self, idx):
         # Convert arrays to PIL images for compatibility with existing transforms
         image = Image.fromarray(image_array)
         label = Image.fromarray(label_array).convert("L")  # Convert RGB to grayscale
-        #     label = Image.fromarray(label_array)
+        # TODO add a check here if it even needs to be converted, to increase perf
 
         # Clean up memory
         del image_array, label_array