Commit 116b014 1 parent 7a1e96c commit 116b014 Copy full SHA for 116b014
File tree 1 file changed +7
-10
lines changed
happypose/pose_estimators/cosypose/cosypose/evaluation
1 file changed +7
-10
lines changed Original file line number Diff line number Diff line change 12
12
from happypose .pose_estimators .cosypose .cosypose .evaluation .prediction_runner import (
13
13
PredictionRunner ,
14
14
)
15
+ from happypose .pose_estimators .cosypose .cosypose .integrated .detector import Detector
15
16
from happypose .pose_estimators .cosypose .cosypose .integrated .pose_estimator import (
16
17
PoseEstimator ,
17
18
)
19
+ from happypose .pose_estimators .cosypose .cosypose .training .detector_models_cfg import (
20
+ check_update_config as check_update_config_detector ,
21
+ )
22
+ from happypose .pose_estimators .cosypose .cosypose .training .detector_models_cfg import (
23
+ create_model_detector ,
24
+ )
18
25
19
26
# Detection
20
27
from happypose .pose_estimators .cosypose .cosypose .training .pose_models_cfg import (
42
49
from happypose .toolbox .utils .distributed import get_rank , get_tmp_dir
43
50
from happypose .toolbox .utils .logging import get_logger
44
51
45
- # """" Temporary imports
46
-
47
52
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
48
53
49
54
logger = get_logger (__name__ )
50
55
51
- from happypose .pose_estimators .cosypose .cosypose .integrated .detector import Detector
52
- from happypose .pose_estimators .cosypose .cosypose .training .detector_models_cfg import (
53
- check_update_config as check_update_config_detector ,
54
- )
55
- from happypose .pose_estimators .cosypose .cosypose .training .detector_models_cfg import (
56
- create_model_detector ,
57
- )
58
-
59
56
60
57
def load_detector (run_id , ds_name ):
61
58
run_dir = EXP_DIR / run_id
You can’t perform that action at this time.
0 commit comments