trainer.py 文件源码

def evaluate_images():
    tb = TB()

    cv2.namedWindow("outputs", cv2.WINDOW_NORMAL)
    test_loader = sLoader.SVT('./svt1/train.xml', './svt1/test.xml')

    while True:
        imgs, anns = test_loader.nextBatch(3,'test')
        pred_labels_f, pred_locs_f, step = tb.sess.run([tb.pred_labels, tb.pred_locs, tb.global_step],
                                                        feed_dict={tb.imgs_ph: imgs, tb.bn: False})
        boxes_, confidences_ = matcher.format_output(pred_labels_f[0], pred_locs_f[0])
        draw_outputs(imgs[0], boxes_, confidences_, wait=0)