Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Security improvements (#10942)

Browse Source 
* Security improvements

* Security improvements

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocherandpre-commit-ci[bot] authored and GitHub committed 2023-02-09 16:57:18 +04:00
1 parent a270b4f125
commit e326252ee4
2 files changed
+7 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

utils/dataloaders.py
+1 -1
View File
@@ -52,7 +52,7 @@ for orientation in ExifTags.TAGS.keys():
def get_hash(paths): def get_hash(paths):
# Returns a single hash value of a list of paths (files or dirs) # Returns a single hash value of a list of paths (files or dirs)
size = sum(os.path.getsize(p) for p in paths if os.path.exists(p)) # sizes size = sum(os.path.getsize(p) for p in paths if os.path.exists(p)) # sizes
h = hashlib.md5(str(size).encode()) # hash sizes h = hashlib.sha256(str(size).encode()) # hash sizes
h.update(''.join(paths).encode()) # hash paths h.update(''.join(paths).encode()) # hash paths
return h.hexdigest() # return hash return h.hexdigest() # return hash
utils/general.py
+6 -5
View File
@@ -14,6 +14,7 @@ import platform
import random import random
import re import re
import signal import signal
import subprocess
import sys import sys
import time import time
import urllib import urllib
@@ -551,7 +552,7 @@ def check_dataset(data, autodownload=True):
r = None # success r = None # success
elif s.startswith('bash '): # bash script elif s.startswith('bash '): # bash script
LOGGER.info(f'Running {s} ...') LOGGER.info(f'Running {s} ...')
r = os.system(s) r = subprocess.run(s, shell=True)
else: # python script else: # python script
r = exec(s, {'yaml': data}) # return None r = exec(s, {'yaml': data}) # return None
dt = f'({round(time.time() - t, 1)}s)' dt = f'({round(time.time() - t, 1)}s)'
@@ -648,9 +649,9 @@ def download(url, dir='.', unzip=True, delete=True, curl=False, threads=1, retry
if is_zipfile(f): if is_zipfile(f):
unzip_file(f, dir) # unzip unzip_file(f, dir) # unzip
elif is_tarfile(f): elif is_tarfile(f):
os.system(f'tar xf {f} --directory {f.parent}') # unzip subprocess.run(f'tar xf {f} --directory {f.parent}', shell=True) # unzip
elif f.suffix == '.gz': elif f.suffix == '.gz':
os.system(f'tar xfz {f} --directory {f.parent}') # unzip subprocess.run(f'tar xfz {f} --directory {f.parent}', shell=True) # unzip
if delete: if delete:
f.unlink() # remove zip f.unlink() # remove zip
@@ -1022,7 +1023,7 @@ def print_mutation(keys, results, hyp, save_dir, bucket, prefix=colorstr('evolve
if bucket: if bucket:
url = f'gs://{bucket}/evolve.csv' url = f'gs://{bucket}/evolve.csv'
if gsutil_getsize(url) > (evolve_csv.stat().st_size if evolve_csv.exists() else 0): if gsutil_getsize(url) > (evolve_csv.stat().st_size if evolve_csv.exists() else 0):
os.system(f'gsutil cp {url} {save_dir}') # download evolve.csv if larger than local subprocess.run(f'gsutil cp {url} {save_dir}', shell=True) # download evolve.csv if larger than local
# Log to evolve.csv # Log to evolve.csv
s = '' if evolve_csv.exists() else (('%20s,' * n % keys).rstrip(',') + '\n') # add header s = '' if evolve_csv.exists() else (('%20s,' * n % keys).rstrip(',') + '\n') # add header
@@ -1046,7 +1047,7 @@ def print_mutation(keys, results, hyp, save_dir, bucket, prefix=colorstr('evolve
for x in vals) + '\n\n') for x in vals) + '\n\n')
if bucket: if bucket:
os.system(f'gsutil cp {evolve_csv} {evolve_yaml} gs://{bucket}') # upload subprocess.run(f'gsutil cp {evolve_csv} {evolve_yaml} gs://{bucket}', shell=True) # upload
def apply_classifier(x, model, img, im0): def apply_classifier(x, model, img, im0):