HRNet / Lite-HRNet

This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.
Apache License 2.0
833 stars 126 forks source link

when load model in test.py flows with error,config with configs/top_down/lite_hrnet/coco/litehrnet_18_coco_256x192.py #4

Closed codylcs closed 3 years ago

codylcs commented 3 years ago

Connected to pydev debugger (build 202.7319.64) Use load_from_local loader The model and loaded state dict do not match exactly

unexpected key in source state_dict: backbone.stage0.2.layers.0.cross_resolution_weighting.conv1.conv.weight, backbone.stage0.2.layers.0.crossresolution...

how to load model in right way?

codylcs commented 3 years ago

run envs

Name Version Build Channel

_libgcc_mutex 0.1 main
absl-py 0.12.0 addict 2.4.0 albumentations 0.5.2 attrs 20.3.0 ca-certificates 2021.4.13 h06a4308_1
cachetools 4.2.1 certifi 2020.12.5 py37h06a4308_0
chardet 4.0.0 chumpy 0.70 click 7.1.2 colorama 0.4.4 coverage 5.5 cudatoolkit 11.0.221 h6bb024c_0
cycler 0.10.0 Cython 0.29.23 dataclasses 0.6 decorator 4.4.2 flake8 3.9.0 flake8-import-order 0.18.1 future 0.18.2 google-auth 1.28.1 google-auth-oauthlib 0.4.4 grpcio 1.37.0 idna 2.10 imageio 2.9.0 imgaug 0.4.0 importlib-metadata 3.10.1 iniconfig 1.1.1 interrogate 1.3.2 isort 4.3.21 json-tricks 3.15.5 kiwisolver 1.3.1 libedit 3.1.20210216 h27cfd23_1
libffi 3.2.1 hf484d3e_1007
libgcc-ng 9.1.0 hdf63c60_0
libstdcxx-ng 9.1.0 hdf63c60_0
Markdown 3.3.4 matplotlib 3.4.1 mccabe 0.6.1 mmcv-full 1.3.0 mmpose 0.13.0 motmetrics 1.2.0 munkres 1.1.4 ncurses 6.2 he6710b0_1
networkx 2.5.1 numpy 1.20.2 oauthlib 3.1.0 onnx 1.8.1 onnxruntime 1.7.0 opencv-python 4.5.1.48 opencv-python-headless 4.5.1.48 openssl 1.0.2u h7b6447c_0
packaging 20.9 pandas 1.1.5 Pillow 8.2.0 pip 21.0.1 py37h06a4308_0
pluggy 0.13.1 poseval 0.1.0 protobuf 3.15.8 py 1.10.0 py-cpuinfo 8.0.0 pyasn1 0.4.8 pyasn1-modules 0.2.8 pycodestyle 2.7.0 pyflakes 2.3.1 pyparsing 2.4.7 pytest 6.2.3 pytest-benchmark 3.2.3 pytest-runner 5.3.0 python 3.7.0 h6e4f718_3
python-dateutil 2.8.1 pytz 2021.1 PyWavelets 1.1.1 PyYAML 5.4.1 readline 7.0 h7b6447c_5
requests 2.25.1 requests-oauthlib 1.3.0 rsa 4.7.2 scikit-image 0.18.1 scipy 1.6.2 setuptools 52.0.0 py37h06a4308_0
Shapely 1.7.1 six 1.15.0 smplx 0.1.26 sqlite 3.33.0 h62c20be_0
tabulate 0.8.9 tensorboard 2.4.1 tensorboard-plugin-wit 1.8.0 tifffile 2021.4.8 tk 8.6.10 hbc83047_0
toml 0.10.2 torch 1.7.0 torchgeometry 0.1.2 torchvision 0.8.1 tqdm 4.60.0 typing-extensions 3.7.4.3 urllib3 1.26.4 Werkzeug 1.0.1 wheel 0.36.2 pyhd3eb1b0_0
xdoctest 0.15.4 xmltodict 0.12.0 xtcocotools 1.7 xz 5.2.5 h7b6447c_0
yapf 0.31.0 zipp 3.4.1 zlib 1.2.11 h7b6447c_3

codylcs commented 3 years ago

sorry to truble u,it slove by match config file with right model weights