Training on dataset which also contain images without annotations gives error.

[teurving]

I'm trying to train the key point RCNN model on a dataset containing images which sometime contain annotations, but also sometime doesn't contain annotations. The annotation file then looks like: {'bboxes':[], 'keypoints':[]}. This gives the error "too many indices for tensor of dimension 1" while trying to calculate the target["area"]. How can I make this code also work for images without annotations?