Skip to content

Commit 456ee9e

Browse files
committed
cleaned config files
1 parent eb60045 commit 456ee9e

File tree

6 files changed

+7
-23
lines changed

6 files changed

+7
-23
lines changed

amadeusgpt/configs/Custom_template.yaml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
keypoint_info:
2-
keypoint_file_path:
3-
model_checkpoint: ""
4-
model_name: ""
2+
keypoint_file_path:
53
body_orientation_keypoints:
64
neck: "nose"
75
tail_base: "tail_base"
@@ -15,5 +13,4 @@ object_info:
1513
load_objects_from_disk: false
1614
video_info:
1715
scene_frame_number: 100
18-
pixels_per_cm: 8
1916
video_file_path:

amadeusgpt/configs/EPM_template.yaml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
keypoint_info:
2-
keypoint_file_path: "examples/EPM/EPM_11DLC_snapshot-1000.h5"
3-
model_checkpoint: ""
4-
model_name: ""
2+
keypoint_file_path: "examples/EPM/EPM_11DLC_snapshot-1000.h5"
53
body_orientation_keypoints:
64
neck: "nose"
75
tail_base: "tail_base"
@@ -15,5 +13,4 @@ llm_info:
1513
keep_last_n_messages: 2
1614
video_info:
1715
scene_frame_number: 100
18-
pixels_per_cm: 8
1916
video_file_path: "examples/EPM/EPM_11DLC_snapshot-1000_labeled_x264.mp4"

amadeusgpt/configs/Horse_template.yaml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
keypoint_info:
22
keypoint_file_path: "examples/Horse/CollectedData_Byron.h5"
3-
model_checkpoint: ""
4-
model_name: ""
53
body_orientation_keypoints:
64
neck: "nose"
75
tail_base: "tail base"

amadeusgpt/configs/MABe_template.yaml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
keypoint_info:
2-
keypoint_file_path: "examples/MABe/mabe_EGS8X2MN4SSUGFWAV976.h5"
3-
model_checkpoint: ""
4-
model_name: ""
2+
keypoint_file_path: "examples/MABe/mabe_EGS8X2MN4SSUGFWAV976.h5"
53
body_orientation_keypoints:
64
neck: "nose"
75
tail_base: "tail base"

amadeusgpt/configs/MausHaus_template.yaml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
keypoint_info:
22
keypoint_file_path: "examples/MausHaus/maushaus_trimmedDLC_snapshot-1000.h5"
3-
model_checkpoint: ""
4-
model_name: ""
53
body_orientation_keypoints:
64
neck: "nose"
75
tail_base: "tail_base"

amadeusgpt/managers/model_manager.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,11 @@
1-
from typing import List
2-
31
from .base import Manager
42

5-
63
class ModelManager(Manager):
4+
"""
5+
Not implemented yet. Perhaps we don't need a model manager.
6+
"""
77
def __init__(self, config):
8-
self.config = config
9-
self.sam_config = config["sam_info"]
10-
self.sa_config = config["keypoint_info"]
11-
self.init_SA_model()
12-
self.init_SAM_model()
8+
self.config = config
139

1410
def init_SA_model(self):
1511
pass

0 commit comments

Comments
 (0)