Revert "Assign label information to heatmap in resultmedia" (#2239)

Songki Choi · web-flow · commit aa5247cfee88 · 2023-06-13T17:53:47.000+09:00
Revert "Assign label information to heatmap in resultmedia (#2220)" This reverts commit 489c8bc.
diff --git a/otx/algorithms/anomaly/adapters/anomalib/callbacks/inference.py b/otx/algorithms/anomaly/adapters/anomalib/callbacks/inference.py
@@ -80,7 +80,6 @@ def on_predict_epoch_end(self, _trainer: pl.Trainer, _pl_module: AnomalyModule,
                 ResultMediaEntity(
                     name="Anomaly Map",
                     type="anomaly_map",
-                    label=dataset_item.annotation_scene.get_labels()[0],
                     annotation_scene=dataset_item.annotation_scene,
                     numpy=(anomaly_map * 255).cpu().numpy().astype(np.uint8),
                 )
diff --git a/otx/algorithms/anomaly/tasks/openvino.py b/otx/algorithms/anomaly/tasks/openvino.py
@@ -196,7 +196,6 @@ def infer(self, dataset: DatasetEntity, inference_parameters: InferenceParameter
             heatmap_media = ResultMediaEntity(
                 name="Anomaly Map",
                 type="anomaly_map",
-                label=label,
                 annotation_scene=dataset_item.annotation_scene,
                 numpy=anomaly_map,
             )

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,6 @@ def on_predict_epoch_end(self, _trainer: pl.Trainer, _pl_module: AnomalyModule,`
`80`	`80`	`ResultMediaEntity(`
`81`	`81`	`name="Anomaly Map",`
`82`	`82`	`type="anomaly_map",`
`83`		`- label=dataset_item.annotation_scene.get_labels()[0],`
`84`	`83`	`annotation_scene=dataset_item.annotation_scene,`
`85`	`84`	`numpy=(anomaly_map * 255).cpu().numpy().astype(np.uint8),`
`86`	`85`	`)`
Original file line number	Diff line number	Diff line change
`@@ -196,7 +196,6 @@ def infer(self, dataset: DatasetEntity, inference_parameters: InferenceParameter`
`196`	`196`	`heatmap_media = ResultMediaEntity(`
`197`	`197`	`name="Anomaly Map",`
`198`	`198`	`type="anomaly_map",`
`199`		`- label=label,`
`200`	`199`	`annotation_scene=dataset_item.annotation_scene,`
`201`	`200`	`numpy=anomaly_map,`
`202`	`201`	`)`