Use pathlib instead of os.path

detect.py

@@ -34,7 +34,7 @@
 import torch.backends.cudnn as cudnn
 
 FILE = Path(__file__).resolve()
-ROOT = FILE.parents[0]  # YOLOv5 root directory
+ROOT = FILE.parent  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative

export.py

@@ -58,7 +58,7 @@
 from torch.utils.mobile_optimizer import optimize_for_mobile
 
 FILE = Path(__file__).resolve()
-ROOT = FILE.parents[0]  # YOLOv5 root directory
+ROOT = FILE.parent  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative

models/tf.py

@@ -19,7 +19,7 @@
 ROOT = FILE.parents[1]  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
-# ROOT = ROOT.relative_to(Path.cwd())  # relative
+# ROOT = ROOT.relative_to(Path.cwd()) # relative
 
 import numpy as np
 import tensorflow as tf

models/yolo.py

@@ -15,7 +15,7 @@
 ROOT = FILE.parents[1]  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
-# ROOT = ROOT.relative_to(Path.cwd())  # relative
+# ROOT = ROOT.relative_to(Path.cwd()) # relative
 
 from models.common import *
 from models.experimental import *

train.py

@@ -33,7 +33,7 @@
 from tqdm import tqdm
 
 FILE = Path(__file__).resolve()
-ROOT = FILE.parents[0]  # YOLOv5 root directory
+ROOT = FILE.parent  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative
@@ -506,7 +506,7 @@ def main(opt, callbacks=Callbacks()):
     # Resume
     if opt.resume and not check_wandb_resume(opt) and not opt.evolve:  # resume an interrupted run
         ckpt = opt.resume if isinstance(opt.resume, str) else get_latest_run()  # specified or most recent path
-        assert os.path.isfile(ckpt), 'ERROR: --resume checkpoint does not exist'
+        assert Path(ckpt).is_file(), 'ERROR: --resume checkpoint does not exist'
         with open(Path(ckpt).parent.parent / 'opt.yaml', errors='ignore') as f:
             opt = argparse.Namespace(**yaml.safe_load(f))  # replace
         opt.cfg, opt.weights, opt.resume = '', ckpt, True  # reinstate

utils/benchmarks.py

@@ -36,7 +36,7 @@
 ROOT = FILE.parents[1]  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
-# ROOT = ROOT.relative_to(Path.cwd())  # relative
+# ROOT = ROOT.relative_to(Path.cwd()) # relative
 
 import export
 import val

utils/datasets.py

@@ -45,7 +45,8 @@
 
 def get_hash(paths):
     # 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
+    _paths = [Path(p) for p in paths]
+    size = sum(p.stat().st_size for p in _paths if p.exists())  # sizes
     h = hashlib.md5(str(size).encode())  # hash sizes
     h.update(''.join(paths).encode())  # hash paths
     return h.hexdigest()  # return hash
@@ -288,7 +289,7 @@ def __init__(self, sources='streams.txt', img_size=640, stride=32, auto=True):
         self.img_size = img_size
         self.stride = stride
 
-        if os.path.isfile(sources):
+        if Path(sources).is_file():
             with open(sources) as f:
                 sources = [x.strip() for x in f.read().strip().splitlines() if len(x.strip())]
         else:
@@ -813,9 +814,8 @@ def collate_fn4(batch):
 # Ancillary functions --------------------------------------------------------------------------------------------------
 def create_folder(path='./new'):
     # Create folder
-    if os.path.exists(path):
-        shutil.rmtree(path)  # delete output folder
-    os.makedirs(path)  # make new output folder
+    shutil.rmtree(path, ignore_errors=True)  # delete output folder
+    Path(path).mkdir(parents=True)  # make new output folder
 
 
 def flatten_recursive(path=DATASETS_DIR / 'coco128'):
@@ -903,7 +903,7 @@ def verify_image_label(args):
                     msg = f'{prefix}WARNING: {im_file}: corrupt JPEG restored and saved'
 
         # verify labels
-        if os.path.isfile(lb_file):
+        if Path(lb_file).is_file():
             nf = 1  # label found
             with open(lb_file) as f:
                 lb = [x.split() for x in f.read().strip().splitlines() if len(x)]

utils/downloads.py

@@ -92,7 +92,7 @@ def gdrive_download(id='16TiPfZj7htmTyhntwcZyEEAejOUxuT6m', file='tmp.zip'):
     # Attempt file download
     out = "NUL" if platform.system() == "Windows" else "/dev/null"
     os.system(f'curl -c ./cookie -s -L "drive.google.com/uc?export=download&id={id}" > {out}')
-    if os.path.exists('cookie'):  # large file
+    if cookie.exists():  # large file
         s = f'curl -Lb ./cookie "drive.google.com/uc?export=download&confirm={get_token()}&id={id}" -o {file}'
     else:  # small file
         s = f'curl -s -L -o {file} "drive.google.com/uc?export=download&id={id}"'

utils/general.py

@@ -804,7 +804,7 @@ def strip_optimizer(f='best.pt', s=''):  # from utils.general import *; strip_op
     for p in x['model'].parameters():
         p.requires_grad = False
     torch.save(x, s or f)
-    mb = os.path.getsize(s or f) / 1E6  # filesize
+    mb = Path(s or f).stat().st_size / 1E6  # filesize
     LOGGER.info(f"Optimizer stripped from {f},{(' saved as %s,' % s) if s else ''} {mb:.1f}MB")
 
 

val.py

@@ -30,7 +30,7 @@
 from tqdm import tqdm
 
 FILE = Path(__file__).resolve()
-ROOT = FILE.parents[0]  # YOLOv5 root directory
+ROOT = FILE.parent  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
     sys.path.append(str(ROOT))  # add ROOT to PATH
 ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative