tests.py 文件源码

def test_pre_process_image():
    matched, debug_gt, debug_matched_default_box, dbg_def_boxes ,dbg_cells, dbg_imgs = pre_process_images("tiny/data.pkl",["img00090526.jpg"])
    img = mpimg.imread("tiny/img00090526.jpg")
    debug_draw_boxes(img, dbg_cells["img00090526.jpg"], (255,255,255),1)
    debug_draw_boxes(img, debug_gt["img00090526.jpg"], (255,0,0),1)
    debug_draw_boxes(img, debug_matched_default_box["img00090526.jpg"], (1,255,1),2)
    debug_draw_boxes(img, dbg_def_boxes["img00090526.jpg"], (1,0,255),1)
    imgplot = plt.imshow(img)
    plt.show()