Spaces:
Build error
Build error
Commit
β’
6940bf0
1
Parent(s):
2b4b353
added different models
Browse files
app.py
CHANGED
@@ -168,7 +168,16 @@ def predict_pipeline(img_input,
|
|
168 |
elif model_input_str == 'full_dog':
|
169 |
path_to_DLCmodel = "models/DLC_Dog_resnet_50_iteration-0_shuffle-0"
|
170 |
pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
171 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
172 |
# read pose cfg as dict
|
173 |
with open(pose_cfg_path, "r") as stream:
|
174 |
pose_cfg_dict = yaml.safe_load(stream)
|
|
|
168 |
elif model_input_str == 'full_dog':
|
169 |
path_to_DLCmodel = "models/DLC_Dog_resnet_50_iteration-0_shuffle-0"
|
170 |
pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
171 |
+
elif model_input_str == 'monkey_face':
|
172 |
+
path_to_DLCmodel = "models/DLC_FacialLandmarks_resnet_50_iteration-1_shuffle-1"
|
173 |
+
pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
174 |
+
elif model_input_str == 'full_human':
|
175 |
+
path_to_DLCmodel = "models/DLC_human_dancing_resnet_101_iteration-0_shuffle-1"
|
176 |
+
pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
177 |
+
elif model_input_str == 'full_monkey':
|
178 |
+
path_to_DLCmodel = "models/DLC_monkey_resnet_50_iteration-0_shuffle-1"
|
179 |
+
pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
180 |
+
|
181 |
# read pose cfg as dict
|
182 |
with open(pose_cfg_path, "r") as stream:
|
183 |
pose_cfg_dict = yaml.safe_load(stream)
|