diff --git a/train_as_chainer.py b/train_as_chainer.py index 6d80ecad89bac6a1f60b778df4a43ca488ae9d08..50c56196e7ee977c930fef6ec322fef1968616e3 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())