From 87cd0d8057fd45151c94f86947ae35e63702f61d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 8 Aug 2022 17:51:28 +0200 Subject: [PATCH] Edge TPU add `--search_delegate` fix (#8902) Resolves https://github.com/ultralytics/yolov5/issues/8842 --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 644e4578b7f0..ab81f7e3ec71 100644 --- a/export.py +++ b/export.py @@ -529,7 +529,7 @@ def export_edgetpu(file, prefix=colorstr('Edge TPU:')): f = str(file).replace('.pt', '-int8_edgetpu.tflite') # Edge TPU model f_tfl = str(file).replace('.pt', '-int8.tflite') # TFLite model - cmd = f"edgetpu_compiler -s -o {file.parent} {f_tfl}" + cmd = f"edgetpu_compiler -s -d -k 10 --out_dir {file.parent} {f_tfl}" subprocess.run(cmd.split(), check=True) LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')