Format with BlackHEADmaster

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())
-