Merge pull request Haochen-Wang409#70 from cjf8899/main

Added "colormap"
ZombaSY · Jul 22, 2022 · cc97b01 · cc97b01
2 parents 4fb3348 + c5f689b
commit cc97b01
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")