From 72e4328c935dd287e8bd7701e387a88a3029785c Mon Sep 17 00:00:00 2001 From: Toru Ogawa Date: Tue, 11 Jul 2017 18:06:16 +0900 Subject: [PATCH] use cupy --- chainercv/utils/bbox/non_maximum_suppression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chainercv/utils/bbox/non_maximum_suppression.py b/chainercv/utils/bbox/non_maximum_suppression.py index 8a8aec51dc..167c745891 100644 --- a/chainercv/utils/bbox/non_maximum_suppression.py +++ b/chainercv/utils/bbox/non_maximum_suppression.py @@ -196,7 +196,7 @@ def _call_nms_kernel(bbox, thresh): mask_dev = cp.zeros((n_bbox * col_blocks,), dtype=np.uint64) bbox = cp.ascontiguousarray(bbox, dtype=np.float32) kern = _load_kernel('nms_kernel', _nms_gpu_code) - kern(blocks, threads, args=(n_bbox, cp.float32(thresh), + kern(blocks, threads, args=(cp.int32(n_bbox), cp.float32(thresh), bbox, mask_dev)) mask_host = mask_dev.get()