Fix EdgeTPU output directory (#7399)
* Fix EdgeTPU output directory Outputs to same directory as --weights * Update export.pypull/7402/head
parent
5333b55e74
commit
2da2466168
|
@ -387,7 +387,7 @@ def export_edgetpu(keras_model, im, 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 {f_tfl}"
|
||||
cmd = f"edgetpu_compiler -s -o {file.parent} {f_tfl}"
|
||||
subprocess.run(cmd, shell=True, check=True)
|
||||
|
||||
LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
|
||||
|
|
Loading…
Reference in New Issue