Skip to content

Unknown model type , exiting! #66

@MosaabRkia

Description

@MosaabRkia

getting this long error
/usr/local/lib/python3.10/dist-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and will be removed in 0.15, please use 'weights' instead.
warnings.warn(
/usr/local/lib/python3.10/dist-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or None for 'weights' are deprecated since 0.13 and will be removed in 0.15. The current behavior is equivalent to passing weights=KeypointRCNN_ResNet50_FPN_Weights.COCO_V1. You can also use weights=KeypointRCNN_ResNet50_FPN_Weights.DEFAULT to get the most up-to-date weights.
warnings.warn(msg)
Downloading: "https://download.pytorch.org/models/keypointrcnn_resnet50_fpn_coco-fc266e95.pth" to /root/.cache/torch/hub/checkpoints/keypointrcnn_resnet50_fpn_coco-fc266e95.pth
100% 226M/226M [00:03<00:00, 78.0MB/s]
/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py:557: UserWarning: This DataLoader will create 8 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
warnings.warn(_create_warning_msg(
Processing with R-CNN: 100% 1/1 [00:04<00:00, 4.10s/it]
2024-05-19 18:15:30.463 | INFO | main:main:241 - Saving results to: /content/expose/output
2024-05-19 18:15:30.468 | WARNING | expose.models.attention.predictor:init:91 - Apply hand network on body: True
2024-05-19 18:15:30.468 | WARNING | expose.models.attention.predictor:init:93 - Apply hand network on hands: True
2024-05-19 18:15:30.469 | WARNING | expose.models.attention.predictor:init:95 - Predict hands: True
2024-05-19 18:15:30.469 | WARNING | expose.models.attention.predictor:init:102 - Predict head: True
2024-05-19 18:15:30.469 | INFO | expose.models.attention.predictor:init:109 - Condition hand on body: True
2024-05-19 18:15:30.469 | INFO | expose.models.attention.predictor:init:112 - Condition hand wrist pose on body: True
2024-05-19 18:15:30.469 | INFO | expose.models.attention.predictor:init:117 - Condition hand finger pose on body: True
2024-05-19 18:15:30.469 | INFO | expose.models.attention.predictor:init:121 - Condition hand shape on body shape: False
2024-05-19 18:15:30.469 | INFO | expose.models.attention.predictor:init:169 - Condition head on body: True
2024-05-19 18:15:30.470 | INFO | expose.models.attention.predictor:init:170 - Condition expression on body: True
2024-05-19 18:15:30.470 | INFO | expose.models.attention.predictor:init:172 - Condition shape on body: False
2024-05-19 18:15:30.470 | INFO | expose.models.attention.predictor:init:173 - Condition neck pose on body: False
2024-05-19 18:15:30.470 | INFO | expose.models.attention.predictor:init:175 - Condition jaw pose on body: True
Traceback (most recent call last):
File "/content/expose/demo.py", line 554, in
main(
File "/usr/local/lib/python3.10/dist-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "/content/expose/demo.py", line 244, in main
model = SMPLXNet(exp_cfg)
File "/content/expose/expose/models/smplx_net.py", line 39, in init
self.smplx = build_attention_head(exp_cfg)
File "/content/expose/expose/models/attention/build.py", line 23, in build_attention_head
return SMPLXHead(cfg)
File "/content/expose/expose/models/attention/predictor.py", line 247, in init
self.body_model = build_body_model(
File "/usr/local/lib/python3.10/dist-packages/smplx/body_models.py", line 2345, in build_layer
raise ValueError(f'Unknown model type {model_type}, exiting!')
ValueError: Unknown model type , exiting!

i have downloaded the checkpoints, and the smplx model too like that

expose/data/models/smplx$ ls
SMPLX_FEMALE.npz SMPLX_MALE.npz SMPLX_NEUTRAL.npz smplx_npz.zip
SMPLX_FEMALE.pkl SMPLX_MALE.pkl SMPLX_NEUTRAL.pkl version.txt
image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions