Spaces:
Sleeping
Sleeping
Miroslav Purkrabek
commited on
Commit
·
ebb18ad
1
Parent(s):
472855f
init models early
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ import spaces
|
|
| 14 |
from twilio.rest import Client
|
| 15 |
|
| 16 |
try:
|
|
|
|
| 17 |
from mmpose.apis import inference_topdown
|
| 18 |
from mmpose.apis import init_model as init_pose_estimator
|
| 19 |
from mmpose.evaluation.functional import nms
|
|
@@ -24,6 +25,8 @@ except (ImportError, ModuleNotFoundError):
|
|
| 24 |
os.system("pip uninstall -y mmpose mmdet mmcv mmengine mmpretrain")
|
| 25 |
os.system("pip install -U openmim")
|
| 26 |
os.system("mim install mmengine mmcv==2.1.0 mmdet==3.3.0 mmpretrain==1.2.0")
|
|
|
|
|
|
|
| 27 |
from mmpose.apis import inference_topdown
|
| 28 |
from mmpose.apis import init_model as init_pose_estimator
|
| 29 |
from mmpose.evaluation.functional import nms
|
|
@@ -176,7 +179,7 @@ class AsyncFrameProcessor:
|
|
| 176 |
self.visualizer = None
|
| 177 |
|
| 178 |
# Initialize models using the standalone function
|
| 179 |
-
|
| 180 |
|
| 181 |
# Start background processing
|
| 182 |
self._start_processing_thread()
|
|
|
|
| 14 |
from twilio.rest import Client
|
| 15 |
|
| 16 |
try:
|
| 17 |
+
import mmcv
|
| 18 |
from mmpose.apis import inference_topdown
|
| 19 |
from mmpose.apis import init_model as init_pose_estimator
|
| 20 |
from mmpose.evaluation.functional import nms
|
|
|
|
| 25 |
os.system("pip uninstall -y mmpose mmdet mmcv mmengine mmpretrain")
|
| 26 |
os.system("pip install -U openmim")
|
| 27 |
os.system("mim install mmengine mmcv==2.1.0 mmdet==3.3.0 mmpretrain==1.2.0")
|
| 28 |
+
|
| 29 |
+
import mmcv
|
| 30 |
from mmpose.apis import inference_topdown
|
| 31 |
from mmpose.apis import init_model as init_pose_estimator
|
| 32 |
from mmpose.evaluation.functional import nms
|
|
|
|
| 179 |
self.visualizer = None
|
| 180 |
|
| 181 |
# Initialize models using the standalone function
|
| 182 |
+
self.det_model, self.pose_model, self.visualizer = init_models()
|
| 183 |
|
| 184 |
# Start background processing
|
| 185 |
self._start_processing_thread()
|