--- a
+++ b/code/6_visualize.py
@@ -0,0 +1,26 @@
+import config
+from utils_evaluation import visualize
+
+# Visualizing patch predictions with overlaid dots.
+print("\n\n+++++ Running 6_visualize.py +++++")
+print("\n----- Visualizing validation set -----")
+# Visualize validation set.
+visualize(wsi_folder=config.args.wsi_val,
+          preds_folder=config.args.preds_val,
+          vis_folder=config.args.vis_val,
+          classes=config.classes,
+          colors=config.colors,
+          num_classes=config.num_classes,
+          patch_size=config.args.patch_size)
+print("----- Finished visualizing validation set -----\n")
+print("----- Visualizing test set -----")
+# Visualize test set.
+visualize(wsi_folder=config.args.wsi_test,
+          preds_folder=config.args.preds_test,
+          vis_folder=config.args.vis_test,
+          classes=config.classes,
+          colors=config.colors,
+          num_classes=config.num_classes,
+          patch_size=config.args.patch_size)
+print("----- Finished visualizing test set -----\n")
+print("+++++ Finished running 6_visualize.py +++++\n\n")