Skip to content

Commit 854d1c4

Browse files
committed
Add none check for sseg saliency map visualization
1 parent 7c73654 commit 854d1c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/python/model_api/visualizer/scene/segmentation/segmentation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def _get_overlays(self, result: ImageResultWithSoftPrediction) -> list[Overlay]:
3737
overlays.append(Overlay(class_map, label=f"Class {i}"))
3838

3939
# Add saliency map
40-
if result.saliency_map.size > 0:
40+
if result.saliency_map is not None and result.saliency_map.size > 0:
4141
saliency_map = cv2.cvtColor(result.saliency_map, cv2.COLOR_BGR2RGB)
4242
overlays.append(Overlay(saliency_map, label="Saliency Map"))
4343

0 commit comments

Comments
 (0)