BGR support in eval

stefanklut · Dec 4, 2023 · e13cec3 · e13cec3
1 parent 2b44f6c
commit e13cec3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/eval.py b/eval.py
@@ -124,7 +124,11 @@ def create_gt_visualization(image_filename, sem_seg_filename):
         def create_pred_visualization(image_filename):
             image = load_image(image_filename)
             logger.info(f"Predict: {image_filename}")
-            outputs = predictor(image)
+            if cfg.INPUT.FORMAT == "BGR":
+                input_image = image[..., ::-1]
+            else:
+                input_image = image
+            outputs = predictor(input_image)
             pred = torch.argmax(outputs[0]["sem_seg"], dim=-3).to("cpu")
             # outputs["panoptic_seg"] = (outputs["panoptic_seg"][0].to("cpu"),
             #                            outputs["panoptic_seg"][1])