Skip to content

Commit

Permalink
Modeld: cannot run without wide img
Browse files Browse the repository at this point in the history
  • Loading branch information
haraschax committed Sep 30, 2024
1 parent 19bd960 commit d4e2572
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions selfdrive/modeld/modeld.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,8 +96,7 @@ def run(self, buf: VisionBuf, wbuf: VisionBuf, transform: np.ndarray, transform_

# if getCLBuffer is not None, frame will be None
self.model.setInputBuffer("input_imgs", self.frame.prepare(buf, transform.flatten(), self.model.getCLBuffer("input_imgs")))
if wbuf is not None:
self.model.setInputBuffer("big_input_imgs", self.wide_frame.prepare(wbuf, transform_wide.flatten(), self.model.getCLBuffer("big_input_imgs")))
self.model.setInputBuffer("big_input_imgs", self.wide_frame.prepare(wbuf, transform_wide.flatten(), self.model.getCLBuffer("big_input_imgs")))

if prepare_only:
return None
Expand Down

0 comments on commit d4e2572

Please sign in to comment.