Updated path handling for yolo .names and .data files -resolves #5

Author: FabianPlum

YOLOader.py

@@ -69,6 +69,24 @@ def create_data(self):
 
         print("Created new .data file for the loaded network.")
 
+
+    def update_data(self):
+        """
+        creates a .data file in the location of the config file with the current location of the .names file and sets
+        the path to the output file as the new .data file.
+        """
+        with open(self.data, "r") as f:
+            lines = f.readlines()
+            for l, line in enumerate(lines):
+                if line.split(" ")[0] == "names":
+                    lines[l] = "names = " + str(self.names).replace("\\", "/") + "\n"
+
+        with open(self.data, "w") as f:
+            for line in lines:
+                f.write(line)
+
+        print("INFO: Updated .data file for the loaded network to include absolute path of .names file")
+
     def update_cfg(self, nw_width, nw_height, encoding="utf-8"):
         """
         Updates the cfg file and creates a copy of it in the same directory as the original. The function updates the

__init__.py

@@ -30,7 +30,7 @@
     "author": "Fabian Plum",
     "description": "Deep learning-based multi animal tracker",
     "blender": (3, 3, 0),
-    "version": (0, 2, 2),
+    "version": (0, 2, 3),
     "location": "",
     "warning": "RUN IN ADMINISTRATOR MODE DURING INSTALLATION!",
     "category": "motion capture"
@@ -262,6 +262,9 @@ def execute(self, context):
 
         if yolo_paths.data is None:
             yolo_paths.create_data()
+        else:
+            # update the data file to ensure it points to the correct absolute location of the names file
+            yolo_paths.update_data()
 
         context.scene.detection_config_path = yolo_paths.cfg
         context.scene.detection_data_path = yolo_paths.data