diff --git a/myeasymocap/backbone/hand2d/hand2d.py b/myeasymocap/backbone/hand2d/hand2d.py index 4cb5ae6..462705a 100644 --- a/myeasymocap/backbone/hand2d/hand2d.py +++ b/myeasymocap/backbone/hand2d/hand2d.py @@ -6,7 +6,6 @@ import numpy as np import math # https://download.openmmlab.com/mmpose/hand/hrnetv2/hrnetv2_w18_rhd2d_256x256-95b20dd8_20210330.pth # https://download.openmmlab.com/mmpose/hand/dark/hrnetv2_w18_onehand10k_256x256_dark-a2f80c64_20210330.pth -from .hrnet import PoseHighResolutionNet from ..basetopdown import BaseTopDownModelCache, get_preds_from_heatmaps, gdown_models class TopDownAsMMPose(nn.Module): @@ -31,6 +30,7 @@ class MyHand2D(BaseTopDownModelCache): def __init__(self, ckpt, url=None, mode='hrnet'): if mode == 'hrnet': super().__init__(name='hand2d', bbox_scale=1.1, res_input=256) + from .hrnet import PoseHighResolutionNet backbone = PoseHighResolutionNet(inp_ch=3, out_ch=21, W=18, multi_scale_final=True, add_final_layer=False) checkpoint = torch.load(ckpt, map_location='cpu')['state_dict'] self.load_checkpoint(backbone, checkpoint, prefix='backbone.', strict=True) diff --git a/requirements.txt b/requirements.txt index ca8cbca..0fd82de 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,8 +5,11 @@ opencv-python yacs tabulate termcolor -chumpy +git+https://github.com/mattloper/chumpy.git mediapipe==0.10.0 func_timeout ultralytics gdown +setuptools==59.5.0 +tensorboard==2.8.0 +pytorch-lightning==1.5.0 \ No newline at end of file