SSD/MobileNet Prediction

ssd_prediction.py

# loop 100 bounding boxes
for detection in Net_SSD_pred[0,0,:,:]:        
    # get the class index
    class_index = int(detection[1])
    # get the score
    score = float(detection[2])
    # draw the bounding box
    if score > threshold:     
        left   = detection[3] * cols
        top    = detection[4] * rows
        right  = detection[5] * cols
        bottom = detection[6] * rows
            
        width  = right - left
        height = bottom - top

        box = patch.Rectangle((int(left), int(top)),
                               width,
                               height,
                               linewidth=2, edgecolor='r', facecolor='none')
                    
        ax1.add_patch(box)