You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Loads checkpoint by local backend from path: work_dirs/hr1/best_coco_AP_epoch_100.pth
/home/dell/anaconda3/envs/mmpose/lib/python3.8/site-packages/mmengine/visualization/visualizer.py:196: UserWarning: Failed to add <class 'mmengine.visualization.vis_backend.LocalVisBackend'>, please provide the save_dir argument.
warnings.warn(f'Failed to add {vis_backend.class}, '
Traceback (most recent call last):
File "demo/image_demo.py", line 106, in
main()
File "demo/image_demo.py", line 87, in main
results = merge_data_samples(batch_results)
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 51, in merge_data_samples
reverted_heatmaps = [
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 52, in
revert_heatmap(data_sample.pred_fields.heatmaps,
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 113, in revert_heatmap
heatmap = heatmap.transpose(2, 0, 1)
ValueError: axes don't match array
Additional information
No response
The text was updated successfully, but these errors were encountered:
Prerequisite
Environment
OrderedDict([('sys.platform', 'linux'), ('Python', '3.8.18 (default, Sep 11 2023, 13:40:15) [GCC 11.2.0]'), ('CUDA available', True), ('MUSA available', False), ('numpy_random_seed', 2147483648), ('GPU 0', 'NVIDIA GeForce RTX 3080'), ('CUDA_HOME', ':/usr/local/cuda'), ('GCC', 'gcc (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0'), ('PyTorch', '1.10.1+cu111'), ('PyTorch compiling details', 'PyTorch built with:\n - GCC 7.3\n - C++ Version: 201402\n - Intel(R) Math Kernel Library Version 2020.0.0 Product Build 20191122 for Intel(R) 64 architecture applications\n - Intel(R) MKL-DNN v2.2.3 (Git Hash 7336ca9f055cf1bfa13efb658fe15dc9b41f0740)\n - OpenMP 201511 (a.k.a. OpenMP 4.5)\n - LAPACK is enabled (usually provided by MKL)\n - NNPACK is enabled\n - CPU capability usage: AVX512\n - CUDA Runtime 11.1\n - NVCC architecture flags: -gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_80,code=sm_80;-gencode;arch=compute_86,code=sm_86\n - CuDNN 8.0.5\n - Magma 2.5.2\n - Build settings: BLAS_INFO=mkl, BUILD_TYPE=Release, CUDA_VERSION=11.1, CUDNN_VERSION=8.0.5, CXX_COMPILER=/opt/rh/devtoolset-7/root/usr/bin/c++, CXX_FLAGS= -Wno-deprecated -fvisibility-inlines-hidden -DUSE_PTHREADPOOL -fopenmp -DNDEBUG -DUSE_KINETO -DUSE_FBGEMM -DUSE_QNNPACK -DUSE_PYTORCH_QNNPACK -DUSE_XNNPACK -DSYMBOLICATE_MOBILE_DEBUG_HANDLE -DEDGE_PROFILER_USE_KINETO -O2 -fPIC -Wno-narrowing -Wall -Wextra -Werror=return-type -Wno-missing-field-initializers -Wno-type-limits -Wno-array-bounds -Wno-unknown-pragmas -Wno-sign-compare -Wno-unused-parameter -Wno-unused-variable -Wno-unused-function -Wno-unused-result -Wno-unused-local-typedefs -Wno-strict-overflow -Wno-strict-aliasing -Wno-error=deprecated-declarations -Wno-stringop-overflow -Wno-psabi -Wno-error=pedantic -Wno-error=redundant-decls -Wno-error=old-style-cast -fdiagnostics-color=always -faligned-new -Wno-unused-but-set-variable -Wno-maybe-uninitialized -fno-math-errno -fno-trapping-math -Werror=format -Wno-stringop-overflow, LAPACK_INFO=mkl, PERF_WITH_AVX=1, PERF_WITH_AVX2=1, PERF_WITH_AVX512=1, TORCH_VERSION=1.10.1, USE_CUDA=ON, USE_CUDNN=ON, USE_EXCEPTION_PTR=1, USE_GFLAGS=OFF, USE_GLOG=OFF, USE_MKL=ON, USE_MKLDNN=ON, USE_MPI=OFF, USE_NCCL=ON, USE_NNPACK=ON, USE_OPENMP=ON, \n'), ('TorchVision', '0.11.2+cu111'), ('OpenCV', '4.9.0'), ('MMEngine', '0.10.3'), ('MMPose', '1.0.0+HEAD')])
Reproduces the problem - code sample
python demo/image_demo.py
/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/data/frame_0319.jpg
data/hr1.py
work_dirs/hr1/best_coco_AP_epoch_100.pth
--out-file vis_results.jpg
--draw-heatmap
Reproduces the problem - command or script
!python demo/image_demo.py
/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/MMPose_Tutorials-main/2023/0524/mmpose/data/frame_0319.jpg
data/hr1.py
work_dirs/hr1/best_coco_AP_epoch_100.pth
--out-file vis_results.jpg
--draw-heatmap
Reproduces the problem - error message
Loads checkpoint by local backend from path: work_dirs/hr1/best_coco_AP_epoch_100.pth
/home/dell/anaconda3/envs/mmpose/lib/python3.8/site-packages/mmengine/visualization/visualizer.py:196: UserWarning: Failed to add <class 'mmengine.visualization.vis_backend.LocalVisBackend'>, please provide the
save_dir
argument.warnings.warn(f'Failed to add {vis_backend.class}, '
Traceback (most recent call last):
File "demo/image_demo.py", line 106, in
main()
File "demo/image_demo.py", line 87, in main
results = merge_data_samples(batch_results)
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 51, in merge_data_samples
reverted_heatmaps = [
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 52, in
revert_heatmap(data_sample.pred_fields.heatmaps,
File "/mnt/cd43550d-ff7b-4058-a185-a6b1f0bd96ea/key/mmpose/mmpose/structures/utils.py", line 113, in revert_heatmap
heatmap = heatmap.transpose(2, 0, 1)
ValueError: axes don't match array
Additional information
No response
The text was updated successfully, but these errors were encountered: