Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ from dlclibrary.dlcmodelzoo.modelzoo_download import (
|
|
21 |
MODELOPTIONS,
|
22 |
)
|
23 |
|
24 |
-
import
|
25 |
|
26 |
# TESTING (passes) download the SuperAnimal models:
|
27 |
#model = 'superanimal_topviewmouse'
|
@@ -93,7 +93,7 @@ def predict_pipeline(img_input,
|
|
93 |
|
94 |
##############################################################
|
95 |
# Run DLC and visualize results
|
96 |
-
dlc_proc = deeplabcut.video_inference_superanimal()
|
97 |
|
98 |
# if required: ignore MD crops and run DLC on full image [mostly for testing]
|
99 |
if flag_dlc_only:
|
|
|
21 |
MODELOPTIONS,
|
22 |
)
|
23 |
|
24 |
+
from dlclive import DLCLive, Processor
|
25 |
|
26 |
# TESTING (passes) download the SuperAnimal models:
|
27 |
#model = 'superanimal_topviewmouse'
|
|
|
93 |
|
94 |
##############################################################
|
95 |
# Run DLC and visualize results
|
96 |
+
dlc_proc = Processor() #TODO: update deeplabcut.video_inference_superanimal() once merged
|
97 |
|
98 |
# if required: ignore MD crops and run DLC on full image [mostly for testing]
|
99 |
if flag_dlc_only:
|