Edge TPU add --search_delegate fix (#8902)

Resolves #8842
ultralytics · Aug 8, 2022 · 4bb3052 · 4bb3052
1 parent c962db2
commit 4bb3052
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export.py b/export.py
@@ -410,7 +410,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)')