cosmetic changes, small bug fixes

examples/COCO_people_subset.ipynb

@@ -66,6 +66,8 @@
     "output_zip = \"../data/COCO_people_subset.zip\"\n",
     "output_folder = \"../data/\"\n",
     "\n",
+    "if not os.path.exists(output_folder):\n",
+    "    os.mkdir(output_folder)\n",
     "# Check if the data already exists\n",
     "if not os.path.exists(output_zip) and not os.path.exists(\n",
     "    os.path.join(output_folder, \"COCO_people_subset\")\n",

luxonis_ml/data/augmentations.py

@@ -14,10 +14,9 @@
     KeypointType,
 )
 
+from luxonis_ml.enums import LabelType
 from luxonis_ml.utils.registry import Registry
 
-from .loader import LabelType
-
 AUGMENTATIONS = Registry(name="augmentations")
 
 

luxonis_ml/data/parsers.py

@@ -22,6 +22,27 @@
 """
 
 
+def parsing_wrapper(func: Callable) -> Callable:
+    """Wrapper for parsing functions that adds data to LDF.
+
+    @type func: Callable
+    @param func: Parsing function
+    @rtype: Callable
+    @return: Wrapper function
+    """
+
+    def wrapper(*args, **kwargs):
+        dataset = args[0].dataset
+        generator, class_names, skeletons, added_images = func(*args, **kwargs)
+        dataset.set_classes(class_names)
+        dataset.set_skeletons(skeletons)
+        dataset.add(generator)
+
+        return added_images
+
+    return wrapper
+
+
 class LuxonisParser:
     def __init__(self, **ldf_kwargs):
         """A parser class used for parsing common dataset formats to LDF.
@@ -36,27 +57,6 @@ def __init__(self, **ldf_kwargs):
         )
         self.dataset = LuxonisDataset(**ldf_kwargs)
 
-    @staticmethod
-    def parsing_wrapper(func: Callable) -> Callable:
-        """Wrapper for parsing functions that adds data to LDF.
-
-        @type func: Callable
-        @param func: Parsing function
-        @rtype: Callable
-        @return: Wrapper function
-        """
-
-        def wrapper(*args, **kwargs):
-            dataset = args[0].dataset
-            generator, class_names, skeletons, added_images = func(*args, **kwargs)
-            dataset.set_classes(class_names)
-            dataset.set_skeletons(skeletons)
-            dataset.add(generator)
-
-            return added_images
-
-        return wrapper
-
     def parse_dir(
         self,
         dataset_type: DatasetType,

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "luxonis-ml"
 version = "0.0.1"
-description = "This package provides MLOps tools for training models for OAK devices"
+description = "MLOps tools for training models for Luxonis devices"
 readme = "README.md"
 requires-python = ">=3.8"
 license = { file = "LICENSE" }
@@ -10,7 +10,7 @@ maintainers = [{ name = "Luxonis", email = "support@luxonis.com" }]
 keywords = ["ml", "ops", "camera", "luxonis", "oak"]
 dynamic = ["dependencies", "optional-dependencies"]
 classifiers = [
-  "License :: Apache License 2.0",
+  "License :: OSI Approved :: Apache Software License",
   "Development Status :: 3 - Alpha",
   "Programming Language :: Python :: 3.8",
   "Programming Language :: Python :: 3.9",
@@ -24,8 +24,8 @@ classifiers = [
 luxonis_ml = "luxonis_ml.luxonis_ml:main"
 
 [project.urls]
-repository = "https://github.com/luxonis/models"
-issues = "https://github.com/luxonis/models/issues"
+repository = "https://github.com/luxonis/luxonis-ml"
+issues = "https://github.com/luxonis/luxonis-ml/issues"
 
 [build-system]
 requires = ["setuptools", "wheel"]