mwmathis commited on
Commit
0da9dca
Β·
1 Parent(s): dc54e8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -10,17 +10,31 @@ import transformers
10
  from PIL import Image
11
  import requests
12
 
 
 
 
 
13
  from deeplabcut.utils import auxiliaryfunctions
14
  from dlclibrary.dlcmodelzoo.modelzoo_download import (
15
  download_huggingface_model,
16
  MODELOPTIONS,
17
  )
18
 
 
 
 
 
 
 
 
 
 
 
19
  model = 'superanimal_topviewmouse'
20
  train_dir = 'DLC_models/sa-tvm'
21
  download_huggingface_model(model, train_dir)
22
 
23
- #demo data cooco cat:
24
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
25
  image = Image.open(requests.get(url, stream=True).raw)
26
 
 
10
  from PIL import Image
11
  import requests
12
 
13
+ from viz_utils import save_results_as_json, draw_keypoints_on_image, draw_bbox_w_text, save_results_only_dlc
14
+ from detection_utils import predict_md, crop_animal_detections
15
+ from ui_utils import gradio_inputs_for_MD_DLC, gradio_outputs_for_MD_DLC, gradio_description_and_examples
16
+
17
  from deeplabcut.utils import auxiliaryfunctions
18
  from dlclibrary.dlcmodelzoo.modelzoo_download import (
19
  download_huggingface_model,
20
  MODELOPTIONS,
21
  )
22
 
23
+ # megadetector and dlc model look up
24
+ MD_models_dict = {'md_v5a': "MD_models/md_v5a.0.0.pt", #
25
+ 'md_v5b': "MD_models/md_v5b.0.0.pt"}
26
+
27
+ # DLC models target dirs
28
+ DLC_models_dict = {'superanimal_topviewmouse': "DLC_models/sa-tvm",
29
+ 'superanimal_quadreped': "DLC_models/sa-q",
30
+ 'full_human': "DLC_models/DLC_human_dancing/"}
31
+
32
+ # download the SuperAnimal models:
33
  model = 'superanimal_topviewmouse'
34
  train_dir = 'DLC_models/sa-tvm'
35
  download_huggingface_model(model, train_dir)
36
 
37
+ # grab demo data cooco cat:
38
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
39
  image = Image.open(requests.get(url, stream=True).raw)
40