search

zju3dv / SMAP

[ECCV 2020] SMAP: Single-Shot Multi-Person Absolute 3D Pose Estimation
Apache License 2.0
241 stars 37 forks source link

Question about MUCO format #37

Closed funnypig521 closed 3 years ago

funnypig521 commented 3 years ago

Excuse me ,I try to train the model and encounter the problem like this: Traceback (most recent call last): File "train.py", line 115, in main() File "train.py", line 55, in main data_loader = get_train_loader(cfg, num_gpu=num_gpu, is_dist=engine.distributed,use_augmentation=True, with_mds=True) File "/home/mist/SMAP/lib/utils/dataloader.py", line 23, in get_train_loader dataset = Dataset(cfg, 'train', transform, use_augmentation, with_mds) File "/home/mist/SMAP/dataset/base_dataset.py", line 43, in init data = data_this['root'] + data KeyError: 'root'

As for coco2017, I find create_annot.py to transform the format,and it works.But for MuCo dataset,how to transform its format? My MuCo dateset is downloaded in your link(https://drive.google.com/drive/folders/1yL2ey3aWHJnh8f_nhWP--IyC9krAPsQN).

raypine commented 3 years ago

follow this link to transform the data format.

funnypig521 commented 3 years ago

Thanks a lot！！！！ @raypine