xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 06:14:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove PNNX debug files (#3677)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
triple Mu 2023-07-12 19:03:37 +08:00 committed by GitHub
parent b239246452
commit 44aa18c99a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/engine/exporter.py
View File

@ -445,6 +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
Path(f).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