Modeld: cannot run without wide img

pull/33678/head
Bruce Wayne 7 months ago
parent 19bd96004e
commit d4e257213d
  1. 3
      selfdrive/modeld/modeld.py

@ -96,8 +96,7 @@ class ModelState:
# if getCLBuffer is not None, frame will be None # 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"))) 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: if prepare_only:
return None return None

Loading…
Cancel
Save