Here’s a thread about how to get a fresh copy of any course notebook.
I added lots of print statements to my code. Here’s what I get when I run the test cell for yolo_eval
:
box_confidence.shape (19, 19, 5, 1)
box_class_probs.shape (19, 19, 5, 80)
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) = 1786
boxes.shape (1786, 4) before scale_boxes
image_shape (720, 1280)
boxes.shape (1786, 4) after scale_boxes
scores[2] = 171.60194
boxes[2] = [-1240.3483 -3212.5881 -645.78 2024.3052]
classes[2] = 16
scores.shape = (10,)
boxes.shape = (10, 4)
classes.shape = (10,)
All tests passed!
Your outputs must be different based on the error message. It looks like perhaps the second dimension of boxes
is 3 instead of 4 in your case, if I’m interpreting your output correctly. That would be a problem if I’m reading that right.