thusiyuan / cooperative_scene_parsing

Code for NeurIPS 2018: Cooperative Holisctic Scene Understanding: Unifying 3D Object, Layout, and Camera Pose Estimation
http://siyuanhuang.com/cooperative_parsing/main.html
MIT License
101 stars 19 forks source link

KeyError: 'seg2d' #18

Open Ivan-VV opened 3 years ago

Ivan-VV commented 3 years ago

Thanks for your great work! I get the error when I run the sunrgbd/sunrgbd_process.py in step 4 of Data. Traceback (most recent call last): File "preprocess/sunrgbd/sunrgbd_process.py", line 666, in <module> main() File "preprocess/sunrgbd/sunrgbd_process.py", line 660, in main prepare_data(False, shift=False) File "preprocess/sunrgbd/sunrgbd_process.py", line 75, in prepare_data sequence = readsunrgbdframe(image_id=i+1) File "/home/data4t/wyf/cooperative_scene_parsing/preprocess/sunrgbd/sunrgbd_parser.py", line 136, in readsunrgbdframe data_frame = SUNRGBDData(img_info['K'], img_info['R_ex'], img_info['R_tilt'], img_info['bdb2d'], img_info['bdb3d'], img_info['gt3dcorner'], img_info['imgdepth'], img_info['imgrgb'], img_info['seg2d'], img_info['sequence_name'], image_id, scene_category) KeyError: 'seg2d'

Then I print the keys of _imginfo and get the result as follows. ['R_ex', 'seg2d_path', 'sensor', 'sequence_name', 'imgrgb_path', 'gt3dcorner', 'R_tilt', 'bdb2d', 'bdb3d', 'K', 'imgdepth_path'] It seems that the dict object _imginfo doesn't have a key called "seg2d". But I have completed the previous three steps according to the instruction Could you please tell me what I should do to solve this problem? Thank you very much!

lx6527562 commented 2 years ago

Hello, I have the same problem, how can I solve it?