[FIX] Squeeze anomaly maps in metadata (#2310)

djdameln · web-flow · commit f94872e7b76b · 2023-07-07T11:44:06.000+09:00
* squeeze anomaly map in meta data
diff --git a/otx/algorithms/anomaly/adapters/anomalib/callbacks/inference.py b/otx/algorithms/anomaly/adapters/anomalib/callbacks/inference.py
@@ -81,7 +81,7 @@ def on_predict_epoch_end(self, _trainer: pl.Trainer, _pl_module: AnomalyModule,
                     name="Anomaly Map",
                     type="anomaly_map",
                     annotation_scene=dataset_item.annotation_scene,
-                    numpy=(anomaly_map * 255).cpu().numpy().astype(np.uint8),
+                    numpy=(anomaly_map * 255).squeeze().cpu().numpy().astype(np.uint8),
                 )
             )
 

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def on_predict_epoch_end(self, _trainer: pl.Trainer, _pl_module: AnomalyModule,`
`81`	`81`	`name="Anomaly Map",`
`82`	`82`	`type="anomaly_map",`
`83`	`83`	`annotation_scene=dataset_item.annotation_scene,`
`84`		`- numpy=(anomaly_map * 255).cpu().numpy().astype(np.uint8),`
	`84`	`+ numpy=(anomaly_map * 255).squeeze().cpu().numpy().astype(np.uint8),`
`85`	`85`	`)`
`86`	`86`	`)`
`87`	`87`