vchiang001 commited on
Commit
6940bf0
β€’
1 Parent(s): 2b4b353

added different models

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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)