help me find a shape error, bellow assertion error
Given how the processing for “boxes” works, the first place to look is your code for filtering_mask, box_class _scores, and box_scores (in that order).
Anytime you get a shape error, the first question is “Ok, what shape is it?” That should then be a clue as to the nature of your problem.
I added a bunch of print statements to my logic and here’s what I see when I run the test cell for yolo_filter_boxes
:
boxes.shape (19, 19, 5, 4)
boxes.dtype <dtype: 'float32'>
box_scores.shape (19, 19, 5, 80)
box_scores.dtype <dtype: 'float32'>
box_classes.shape (19, 19, 5)
box_classes.dtype <dtype: 'int64'>
box_class_scores.shape (19, 19, 5)
box_class_scores.dtype <dtype: 'float32'>
filtering_mask.shape (19, 19, 5)
filtering_mask.dtype <dtype: 'bool'>
sum(filtering_mask) = 1789
scores[2] = 9.270486
boxes[2] = [ 4.6399336 3.2303846 4.431282 -2.202031 ]
classes[2] = 8
scores.shape = (1789,)
boxes.shape = (1789, 4)
classes.shape = (1789,)
All tests passed!
Good hint Paul & Tom that did it
Thank you
Franklin Viloria