zejunyang commited on
Commit
938daee
1 Parent(s): 96ea347
lite_openpose/body_bbox_detector.py CHANGED
@@ -14,7 +14,7 @@ import torchvision.transforms as transforms
14
  # Code from https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch/blob/master/demo.py
15
 
16
  # 2D body pose estimator
17
- sys.path.append('/apdcephfs/share_1474453/zejunzhang/workspace/HR-VITON/dataset_process_utils/lite_openpose')
18
  from pose2d_models.with_mobilenet import PoseEstimationWithMobileNet
19
  from modules.load_state import load_state
20
  from modules.pose import Pose, track_poses
 
14
  # Code from https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch/blob/master/demo.py
15
 
16
  # 2D body pose estimator
17
+ sys.path.append('lite_openpose')
18
  from pose2d_models.with_mobilenet import PoseEstimationWithMobileNet
19
  from modules.load_state import load_state
20
  from modules.pose import Pose, track_poses