Pārlūkot izejas kodu

Fix EdgeTPU output directory (#7399)

* Fix EdgeTPU output directory

Outputs to same directory as --weights

* Update export.py
modifyDataloader
Glenn Jocher GitHub pirms 2 gadiem
vecāks
revīzija
2da2466168
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      export.py

+ 1
- 1
export.py Parādīt failu

@@ -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)')

Notiek ielāde…
Atcelt
Saglabāt