Skip to content

Commit 7c9f989

Browse files
authored
Merge pull request #1025 from luxonis/fix-nnet-callbacks
Check if visualizer is enabled before trying to set frame_shape
2 parents 153de22 + f7e467c commit 7c9f989

File tree

1 file changed

+2
-1
lines changed
  • depthai_sdk/src/depthai_sdk/oak_outputs/xout

1 file changed

+2
-1
lines changed

depthai_sdk/src/depthai_sdk/oak_outputs/xout/xout_nn.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,8 @@ def on_callback(self, packet: Union[DetectionPacket, TrackerPacket]):
9999
if self._frame_shape is None:
100100
# Lazy-load the frame shape
101101
self._frame_shape = np.array([packet.frame.shape[0], packet.frame.shape[1]])
102-
self._visualizer.frame_shape = self._frame_shape
102+
if self._visualizer:
103+
self._visualizer.frame_shape = self._frame_shape
103104

104105
bbox = BoundingBox().resize_to_aspect_ratio(packet.frame.shape, self._nn_size, self._resize_mode)
105106

0 commit comments

Comments
 (0)