Pass all tests by the added binary segmentation exameple

Author: omidvarnia

Makefile

@@ -1,26 +1,39 @@
-.PHONY: test
+.PHONY: test  # Declare the 'test' target as phony to avoid conflicts with files named 'test'
 
+# Variables to store the paths of the python, pip, pytest, and ruff executables
+PYTHON := $(shell which python)
+PIP := $(shell which pip)
+PYTEST := $(shell which pytest)
+RUFF := $(shell which ruff)
+
+# Target to create a Python virtual environment
 .venv:
-	python3 -m venv .venv
+	$(PYTHON) -m venv $(shell dirname $(PYTHON))
 
+# Target to install development dependencies in the virtual environment
 install_dev: .venv
-	.venv/bin/pip install -e ".[test]"
+	$(PIP) install -e ".[test]"
 
+# Target to run tests with pytest, using 2 parallel processes and only non-marked tests
 test: .venv
-	.venv/bin/pytest -v -rsx -n 2 tests/ --non-marked-only
+	$(PYTEST) -v -rsx -n 2 tests/ --non-marked-only
 
+# Target to run all tests with pytest, including slow tests, using 2 parallel processes
 test_all: .venv
-	RUN_SLOW=1 .venv/bin/pytest -v -rsx -n 2 tests/
+	RUN_SLOW=1 $(PYTEST) -v -rsx -n 2 tests/
 
+# Target to generate a table by running a Python script
 table:
-	.venv/bin/python misc/generate_table.py
+	$(PYTHON) misc/generate_table.py
 
+# Target to generate a table for timm by running a Python script
 table_timm:
-	.venv/bin/python misc/generate_table_timm.py
+	$(PYTHON) misc/generate_table_timm.py
 
+# Target to fix and format code using ruff
 fixup:
-	.venv/bin/ruff check --fix
-	.venv/bin/ruff format
+	$(RUFF) check --fix
+	$(RUFF) format
 
+# Target to run code formatting and tests
 all: fixup test
-

examples/binary_segmentation_buildings.py

@@ -72,7 +72,9 @@
 # ----------------------------
 # Download the CamVid dataset, if needed
 # ----------------------------
-main_dir = "./examples/binary_segmentation_data/"  # Change this to your desired directory
+main_dir = (
+    "./examples/binary_segmentation_data/"  # Change this to your desired directory
+)
 
 data_dir = os.path.join(main_dir, "dataset")
 if not os.path.exists(data_dir):
@@ -246,7 +248,6 @@ def visualize(output_dir, image_filename, **images):
 def train_and_evaluate_one_epoch(
     model, train_dataloader, valid_dataloader, optimizer, scheduler, loss_fn, device
 ):
-    
     # Set the model to training mode
     model.train()
     train_loss = 0
@@ -321,7 +322,6 @@ def train_model(
 
 
 def test_model(model, output_dir, test_dataloader, loss_fn, device):
-    
     # Set the model to evaluation mode
     model.eval()
     test_loss = 0