Update eval.py

ZombaSY · Jul 22, 2022 · c5f689b · c5f689b
1 parent 4fb3348
commit c5f689b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval.py b/eval.py
@@ -350,7 +350,7 @@ def valiadte_whole(
         check_makedirs(gray_folder)
         check_makedirs(color_folder)
         gray = np.uint8(prediction)
-        color = colorize(gray)
+        color = colorize(gray,colormap)
         image_path, _ = data_list[i]
         image_name = image_path.split("/")[-1].split(".")[0]
         gray_path = os.path.join(gray_folder, image_name + ".png")