tests.py 文件源码

def test_process_image_set(dirname):
    matched, debug_gt, debug_matched_default_box, dbg_def_boxes ,dbg_cells,dbg_imgs = pre_process_images(dirname+"/data.pkl")

    for img_name in dbg_imgs:
        print("loading ... ",img_name)
        img = mpimg.imread(img_name)
        # debug_draw_boxes(img, dbg_cells[img_name], (255,255,255),1)
        debug_draw_boxes(img, debug_gt[img_name], (255,0,0),1)
        debug_draw_boxes(img, debug_matched_default_box[img_name], (1,255,1),1)
        # debug_draw_boxes(img, dbg_def_boxes[img_name], (1,0,255),1)
        imgplot = plt.imshow(img)
        plt.show()