open-mmlab / mmyolo

OpenMMLab YOLO series toolbox and benchmark. Implemented RTMDet, RTMDet-Rotated,YOLOv5, YOLOv6, YOLOv7, YOLOv8,YOLOX, PPYOLOE, etc.
https://mmyolo.readthedocs.io/zh_CN/dev/
GNU General Public License v3.0
3.02k stars 544 forks source link

run yolox_pose image_demo.py KeyError: 'flip_indices' #900

Open betterhalfwzm opened 1 year ago

betterhalfwzm commented 1 year ago

Prerequisite

🐞 Describe the bug

File "/home/wangzhaoming/train_code/mmyolo/demo/images_demo.py", line 189, in main() File "/home/wangzhaoming/train_code/mmyolo/demo/images_demo.py", line 128, in main result = inference_detector(model, file) File "/home/wangzhaoming/anaconda3/envs/TTC_OD/lib/python3.10/site-packages/mmdet/apis/inference.py", line 182, in inferencedetector data = testpipeline(data) File "/home/wangzhaoming/anaconda3/envs/TTC_OD/lib/python3.10/site-packages/mmcv/transforms/base.py", line 12, in call return self.transform(results) File "/home/wangzhaoming/anaconda3/envs/TTC_OD/lib/python3.10/site-packages/mmcv/transforms/wrappers.py", line 88, in transform results = t(results) # type: ignore File "/home/wangzhaoming/anaconda3/envs/TTC_OD/lib/python3.10/site-packages/mmcv/transforms/base.py", line 12, in call return self.transform(results) File "/home/wangzhaoming/train_code/mmyolo/mmyolo/datasets/transforms/transforms.py", line 444, in transform self._load_kps(results) File "/home/wangzhaoming/train_code/mmyolo/mmyolo/datasets/transforms/transforms.py", line 640, in _load_kps (0, len(results['flip_indices']), 2), dtype=np.float32) KeyError: 'flip_indices'

Environment

#

Name Version Build Channel

_libgcc_mutex 0.1 main defaults _openmp_mutex 5.1 1_gnu defaults addict 2.4.0 pypi_0 pypi albumentations 1.3.1 pypi_0 pypi aliyun-python-sdk-core 2.13.36 pypi_0 pypi aliyun-python-sdk-kms 2.16.2 pypi_0 pypi bzip2 1.0.8 h7b6447c_0 defaults ca-certificates 2023.08.22 h06a4308_0 defaults certifi 2023.7.22 pypi_0 pypi cffi 1.15.1 pypi_0 pypi charset-normalizer 3.2.0 pypi_0 pypi chumpy 0.70 pypi_0 pypi click 8.1.7 pypi_0 pypi cmake 3.27.5 pypi_0 pypi colorama 0.4.6 pypi_0 pypi contourpy 1.1.1 pypi_0 pypi crcmod 1.7 pypi_0 pypi cryptography 41.0.4 pypi_0 pypi cycler 0.11.0 pypi_0 pypi cython 3.0.3 pypi_0 pypi filelock 3.12.4 pypi_0 pypi fonttools 4.42.1 pypi_0 pypi idna 3.4 pypi_0 pypi imageio 2.31.3 pypi_0 pypi importlib-metadata 6.8.0 pypi_0 pypi jinja2 3.1.2 pypi_0 pypi jmespath 0.10.0 pypi_0 pypi joblib 1.3.2 pypi_0 pypi json-tricks 3.17.3 pypi_0 pypi kiwisolver 1.4.5 pypi_0 pypi lazy-loader 0.3 pypi_0 pypi ld_impl_linux-64 2.38 h1181459_1 defaults libffi 3.4.4 h6a678d5_0 defaults libgcc-ng 11.2.0 h1234567_1 defaults libgomp 11.2.0 h1234567_1 defaults libstdcxx-ng 11.2.0 h1234567_1 defaults libuuid 1.41.5 h5eee18b_0 defaults lit 16.0.6 pypi_0 pypi markdown 3.4.4 pypi_0 pypi markdown-it-py 3.0.0 pypi_0 pypi markupsafe 2.1.3 pypi_0 pypi matplotlib 3.8.0 pypi_0 pypi mdurl 0.1.2 pypi_0 pypi mmcv 2.0.1 pypi_0 pypi mmdet 3.1.0 pypi_0 pypi mmengine 0.8.4 pypi_0 pypi mmpose 1.1.0 pypi_0 pypi mmsegmentation 1.1.2 dev_0 mmyolo 0.6.0 dev_0 model-index 0.1.11 pypi_0 pypi mpmath 1.3.0 pypi_0 pypi munkres 1.1.4 pypi_0 pypi ncurses 6.4 h6a678d5_0 defaults networkx 3.1 pypi_0 pypi numpy 1.26.0 pypi_0 pypi opencv-python 4.8.0.76 pypi_0 pypi opendatalab 0.0.10 pypi_0 pypi openmim 0.3.9 pypi_0 pypi openssl 3.0.10 h7f8727e_2 defaults openxlab 0.0.25 pypi_0 pypi ordered-set 4.1.0 pypi_0 pypi oss2 2.17.0 pypi_0 pypi packaging 23.1 pypi_0 pypi pandas 2.1.1 pypi_0 pypi pillow 10.0.1 pypi_0 pypi pip 23.2.1 py310h06a4308_0 defaults platformdirs 3.10.0 pypi_0 pypi prettytable 3.9.0 pypi_0 pypi pycocotools 2.0.7 pypi_0 pypi pycparser 2.21 pypi_0 pypi pycryptodome 3.19.0 pypi_0 pypi pygments 2.16.1 pypi_0 pypi pyparsing 3.1.1 pypi_0 pypi python 3.10.13 h955ad1f_0 defaults python-dateutil 2.8.2 pypi_0 pypi pytz 2023.3.post1 pypi_0 pypi pywavelets 1.4.1 pypi_0 pypi pyyaml 6.0.1 pypi_0 pypi qudida 0.0.4 pypi_0 pypi readline 8.2 h5eee18b_0 defaults requests 2.28.2 pypi_0 pypi rich 13.4.2 pypi_0 pypi scikit-image 0.21.0 pypi_0 pypi scikit-learn 1.3.1 pypi_0 pypi scipy 1.11.2 pypi_0 pypi setuptools 60.2.0 pypi_0 pypi shapely 2.0.1 pypi_0 pypi six 1.16.0 pypi_0 pypi sqlite 3.41.2 h5eee18b_0 defaults sympy 1.12 pypi_0 pypi tabulate 0.9.0 pypi_0 pypi termcolor 2.3.0 pypi_0 pypi terminaltables 3.1.10 pypi_0 pypi threadpoolctl 3.2.0 pypi_0 pypi tifffile 2023.9.18 pypi_0 pypi tk 8.6.12 h1ccaba5_0 defaults tomli 2.0.1 pypi_0 pypi torch 2.0.1+cu118 pypi_0 pypi torchvision 0.15.2+cu118 pypi_0 pypi tqdm 4.65.2 pypi_0 pypi triton 2.0.0 pypi_0 pypi typing-extensions 4.8.0 pypi_0 pypi tzdata 2023.3 pypi_0 pypi urllib3 1.26.16 pypi_0 pypi wcwidth 0.2.6 pypi_0 pypi wheel 0.38.4 py310h06a4308_0 defaults xtcocotools 1.13 pypi_0 pypi xz 5.4.2 h5eee18b_0 defaults yapf 0.40.1 pypi_0 pypi zipp 3.17.0 pypi_0 pypi zlib 1.2.13 h5eee18b_0 defaults

Additional information

yes

winterxx commented 1 year ago

I meet the same problem

Jackqu commented 11 months ago

I also meet this problem

Howard9112 commented 10 months ago

I also have this problem

ruhyadi commented 7 months ago

Any fix for this problems?