Merge pull request #123 from ttngu207/main_external-storage

feat: store output files from `analyze_videos()` as `filepath`

element_deeplabcut/model.py

@@ -704,6 +704,14 @@ class BodyPartPosition(dj.Part):
         likelihood  : longblob
         """
 
+    class File(dj.Part):
+        definition = """
+        -> master
+        file_name: varchar(255)
+        ---
+        file: filepath@dlc-processed
+        """
+
     def make(self, key):
         """.populate() method will launch training for each PoseEstimationTask"""
         # ID model and directories
@@ -835,6 +843,18 @@ def do_analyze_videos():
 
         self.insert1({**key, "pose_estimation_time": creation_time})
         self.BodyPartPosition.insert(body_parts)
+        # Insert result files
+        self.File.insert(
+            [
+                {
+                    **key,
+                    "file_name": f.relative_to(output_dir).as_posix(),
+                    "file": f,
+                }
+                for f in output_dir.rglob("*")
+                if f.is_file()
+            ]
+        )
 
     @classmethod
     def get_trajectory(cls, key: dict, body_parts: list = "all") -> pd.DataFrame: