HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix NCNN multiple-volumes PNNX download bug (#14533)

Browse source
This commit is contained in:
Ariel Kukulanski 2024-07-18 13:22:33 -03:00 • committed by GitHub
parent f50274c596
commit 0822710185
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/exporter.py
View file

@ -563,7 +563,7 @@ class Exporter:
LOGGER.warning(f"{prefix} WARNING ⚠️ PNNX GitHub assets not found: {e}, using default {asset}") LOGGER.warning(f"{prefix} WARNING ⚠️ PNNX GitHub assets not found: {e}, using default {asset}")
unzip_dir = safe_download(f"https://github.com/pnnx/pnnx/releases/download/{release}/{asset}", delete=True) unzip_dir = safe_download(f"https://github.com/pnnx/pnnx/releases/download/{release}/{asset}", delete=True)
if check_is_path_safe(Path.cwd(), unzip_dir): # avoid path traversal security vulnerability if check_is_path_safe(Path.cwd(), unzip_dir): # avoid path traversal security vulnerability
(unzip_dir / name).rename(pnnx) # move binary to ROOT shutil.move(src=unzip_dir / name, dst=pnnx) # move binary to ROOT
pnnx.chmod(0o777) # set read, write, and execute permissions for everyone pnnx.chmod(0o777) # set read, write, and execute permissions for everyone
shutil.rmtree(unzip_dir) # delete unzip dir shutil.rmtree(unzip_dir) # delete unzip dir