From 50f8eaf973e8fb7092d9508b3563c8fdaea12d35 Mon Sep 17 00:00:00 2001 From: chenyuntc Date: Thu, 21 Dec 2017 10:47:48 +0800 Subject: [PATCH] tiny fix of chainer training --- train_as_chainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_as_chainer.py b/train_as_chainer.py index 6d80eca..50c5619 100644 --- a/train_as_chainer.py +++ b/train_as_chainer.py @@ -23,7 +23,7 @@ def eval(dataloader, faster_rcnn, test_num=10000): gt_bboxes, gt_labels, gt_difficults = list(), list(), list() for ii, (imgs, sizes, gt_bboxes_, gt_labels_, gt_difficults_) in tqdm(enumerate(dataloader)): sizes = [sizes[0][0], sizes[1][0]] - pred_bboxes_, pred_labels_, pred_scores_ = faster_rcnn.predict(imgs, [sizes]) + pred_bboxes_, pred_labels_, pred_scores_ = faster_rcnn.predict2(imgs, [sizes]) gt_bboxes += list(gt_bboxes_.numpy()) gt_labels += list(gt_labels_.numpy()) gt_difficults += list(gt_difficults_.numpy()) -- GitLab