|
|
@ -445,8 +445,8 @@ class Exporter:
|
|
|
|
f.mkdir(exist_ok=True) # make ncnn_model directory
|
|
|
|
f.mkdir(exist_ok=True) # make ncnn_model directory
|
|
|
|
LOGGER.info(f"{prefix} running '{' '.join(cmd)}'")
|
|
|
|
LOGGER.info(f"{prefix} running '{' '.join(cmd)}'")
|
|
|
|
subprocess.run(cmd, check=True)
|
|
|
|
subprocess.run(cmd, check=True)
|
|
|
|
for f in 'debug.bin', 'debug.param', 'debug2.bin', 'debug2.param': # remove debug files
|
|
|
|
for f_debug in 'debug.bin', 'debug.param', 'debug2.bin', 'debug2.param': # remove debug files
|
|
|
|
Path(f).unlink(missing_ok=True)
|
|
|
|
Path(f_debug).unlink(missing_ok=True)
|
|
|
|
|
|
|
|
|
|
|
|
yaml_save(f / 'metadata.yaml', self.metadata) # add metadata.yaml
|
|
|
|
yaml_save(f / 'metadata.yaml', self.metadata) # add metadata.yaml
|
|
|
|
return str(f), None
|
|
|
|
return str(f), None
|
|
|
|