sfmig commited on
Commit
8dab660
·
1 Parent(s): b761da9

adding cat pb

Browse files
DLC_models/DLC_Cat_resnet_50_iteration-0_shuffle-0/snapshot-75000.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d44893d1d7a0a07afd37b28c8c00cd8178ae69a0ff49152f16c08c53b8efa4da
3
+ size 98887790
app.py CHANGED
@@ -116,6 +116,7 @@ def predict_dlc(list_np_crops,DLCmodel,dlc_proc):
116
 
117
  list_kpts_per_crop = []
118
  for crop in list_np_crops:
 
119
  keypts = dlc_live.get_pose(crop) # third column is llk!
120
  list_kpts_per_crop.append(keypts)
121
 
@@ -145,11 +146,11 @@ def predict_pipeline(img_input):
145
 
146
 
147
  # # Produce final image
148
- fig = plt.Figure(md_results.imgs[0].shape[:2]) #figsize=(10,10)) #md_results.imgs[0].shape)
149
  for ic, (np_crop, kpts_crop) in enumerate(zip(list_crops,
150
  list_kpts_per_crop)):
151
 
152
- # Draw keypts on crop
153
  img_crop = Image.fromarray(np_crop)
154
  draw_keypoints_on_image(img_crop,
155
  kpts_crop, # a numpy array with shape [num_keypoints, 2].
@@ -157,9 +158,8 @@ def predict_pipeline(img_input):
157
  radius=2,
158
  use_normalized_coordinates=False) # if True, then I should use md_results.xyxyn
159
 
160
- # Paste crop in original image
161
  # https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.paste
162
-
163
  img_input.paste(img_crop,
164
  box = tuple([int(math.floor(t)) for t in md_results.xyxy[0][ic,:2]]))
165
 
 
116
 
117
  list_kpts_per_crop = []
118
  for crop in list_np_crops:
119
+ # scale crop?
120
  keypts = dlc_live.get_pose(crop) # third column is llk!
121
  list_kpts_per_crop.append(keypts)
122
 
 
146
 
147
 
148
  # # Produce final image
149
+ # fig = plt.Figure(md_results.imgs[0].shape[:2]) #figsize=(10,10)) #md_results.imgs[0].shape)
150
  for ic, (np_crop, kpts_crop) in enumerate(zip(list_crops,
151
  list_kpts_per_crop)):
152
 
153
+ ## Draw keypts on crop
154
  img_crop = Image.fromarray(np_crop)
155
  draw_keypoints_on_image(img_crop,
156
  kpts_crop, # a numpy array with shape [num_keypoints, 2].
 
158
  radius=2,
159
  use_normalized_coordinates=False) # if True, then I should use md_results.xyxyn
160
 
161
+ ## Paste crop in original image
162
  # https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.paste
 
163
  img_input.paste(img_crop,
164
  box = tuple([int(math.floor(t)) for t in md_results.xyxy[0][ic,:2]]))
165