|
|
|
@ -96,8 +96,7 @@ class ModelState: |
|
|
|
|
|
|
|
|
|
# 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 |
|
|
|
|