octiapp / KerasPersonLab

Keras-tensorflow implementation of PersonLab (https://arxiv.org/abs/1803.08225)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error when training!

Minotaur-CN opened this issue · comments

hi,all
One error happens when training from scratch.


KerasPersonLab-updated_model_def/data_prep.py", line 16, in get_ground_truth

assert(instance_masks.shape[-1] == len(all_keypoints)), '{} != {}'.format(instance_masks.shape[-1], len(all_keypoints))

AssertionError: 7 != 8

Hello @Minotaur-CN @nitzan1207
The cause of this error was a COCO segm annotation with zero area. I updated the generate_hdf5 script to handle this, so you will need to delete your current hdf5 file and re-run the script again.