diff options
Diffstat (limited to 'YOSO/__init__.py')
-rw-r--r-- | YOSO/__init__.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/YOSO/__init__.py b/YOSO/__init__.py index 1decc61..6798d6f 100644 --- a/YOSO/__init__.py +++ b/YOSO/__init__.py @@ -5,24 +5,30 @@ import re import sys -IMAGE_FILE_TEMPLATES = ['*.png', '*.jpg', '*.jpeg'] +IMAGE_FILE_TEMPLATES = ["*.png", "*.jpg", "*.jpeg"] + def imagesDir(datadir): - return os.path.join(datadir, 'images') + return os.path.join(datadir, "images") + def labelsDir(datadir): - return os.path.join(datadir, 'labels') + return os.path.join(datadir, "labels") + def classesDir(datadir): - return os.path.join(datadir, 'classes') + return os.path.join(datadir, "classes") + # e. g. "012 - Midi skirt.jpg": -CLASSES_RE = re.compile(r'^0*(?P<cls>\d+)\s*-\s*(?P<name>[^.]+).*$') +CLASSES_RE = re.compile(r"^0*(?P<cls>\d+)\s*-\s*(?P<name>[^.]+).*$") # e. g. "1 0.57 0.42 0.17 0.6654" BBOX_RE = re.compile( - r'^\s*(?P<cls>\d+)\s+(?P<x>{float})\s+(?P<y>{float})\s+(?P<w>{float})\s+(?P<h>{float}).*$'.format( - float=r'([0-9]*[.])?[0-9]+')) + r"^\s*(?P<cls>\d+)\s+(?P<x>{float})\s+(?P<y>{float})\s+(?P<w>{float})\s+(?P<h>{float}).*$".format( + float=r"([0-9]*[.])?[0-9]+" + ) +) def main(): @@ -32,4 +38,3 @@ def main(): main_window.show() sys.exit(app.exec()) - |