Closed chanangaza closed 3 years ago
def load_tflite(
self, tflite_path: str, edgetpu_lib: str = EDGETPU_SHARED_LIB
) -> None:
if self.tpu:
self.interpreter = tflite.Interpreter(
model_path=tflite_path,
experimental_delegates=[load_delegate(edgetpu_lib)],
)
else:
self.interpreter = tflite.Interpreter(model_path=tflite_path)
How about this?
I don't have a USB accelerator. Can you test your code on Windows or Mac?
Please make PR if possible. :)
def load_tflite( self, tflite_path: str, edgetpu_lib: str = EDGETPU_SHARED_LIB ) -> None: if self.tpu: self.interpreter = tflite.Interpreter( model_path=tflite_path, experimental_delegates=[load_delegate(edgetpu_lib)], ) else: self.interpreter = tflite.Interpreter(model_path=tflite_path)
How about this?
~I don't have a USB accelerator. Can you test your code on Windows or Mac?~
Please make PR if possible. :)
this is even better;) already tested in on windows with google-coral edge-tpu USB accelerator.
First, I think this is an amazing repository. Well documented and designed.
I wish to run yolov4 with TPU on windows however, the code only supports linux runtime names.
thanks in advance and please let me know whether I have overlooked something here...