search

DirtyHarryLYL / HAKE-Action

As a part of the HAKE project, includes the reproduced SOTA models and the corresponding HAKE-enhanced versions (CVPR2020).
Apache License 2.0
100 stars 13 forks source link

Something goes wrong when training PaStaNet in HAKE-Large model! #53

Closed Shunli-Wang closed 4 years ago

Shunli-Wang commented 4 years ago

When I run this training command: python tools/Train_pasta_HICO_DET.py --data 1 --init_weight 1 --train_module 2 --num_iteration 2000000 --model _pasta_large_pretrain It turns out that the data loading function is still Augmented_HO_Neg_HICO_DET_for_only_PVP76, and the GT&Neg 10w.pkl has totally different data structure with Trainval_GT_all_part.pkl and Trainval_Neg_all_part.pkl. Is there anything wrong with my code? Looking forward to your replay!

Here is the error: File "/home/fair/Desktop/wsl/_HOI/Instance_HAKE/tools/../lib/ult/ult_HICO_DET.py", line 433, in Augmented_HO_Neg_HICO_DET_for_only_PVP76 part_bbox.append(GT[i][4]['part_bbox'][None, :, :]) # get part boxes (10,5) TypeError: string indices must be integers