Browse Source
Fix EdgeTPU output directory (#7399)
* Fix EdgeTPU output directory Outputs to same directory as --weights * Update export.pymodifyDataloader
Glenn Jocher
GitHub
2 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
export.py
View File
| f = str(file).replace('.pt', '-int8_edgetpu.tflite') # Edge TPU model | f = str(file).replace('.pt', '-int8_edgetpu.tflite') # Edge TPU model | ||||
| f_tfl = str(file).replace('.pt', '-int8.tflite') # TFLite 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) | subprocess.run(cmd, shell=True, check=True) | ||||
| LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') | LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') |
Loading…