Skip to content

Commit b1c1aaf

Browse files
authored
Fix EdgeTPU output directory (ultralytics#7399)
* Fix EdgeTPU output directory Outputs to same directory as --weights * Update export.py
1 parent 5c4587f commit b1c1aaf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ def export_edgetpu(keras_model, im, file, prefix=colorstr('Edge TPU:')):
387387
f = str(file).replace('.pt', '-int8_edgetpu.tflite') # Edge TPU model
388388
f_tfl = str(file).replace('.pt', '-int8.tflite') # TFLite model
389389

390-
cmd = f"edgetpu_compiler -s {f_tfl}"
390+
cmd = f"edgetpu_compiler -s -o {file.parent} {f_tfl}"
391391
subprocess.run(cmd, shell=True, check=True)
392392

393393
LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')

0 commit comments

Comments
 (0)