From bf05e25a11b420649f1c3f8bc58af66ba7bdf5c8 Mon Sep 17 00:00:00 2001 From: yangxue Date: Tue, 24 Nov 2020 17:19:20 +0800 Subject: [PATCH] Update test_coco_multiprocessing.py --- tools/test_coco_multiprocessing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/test_coco_multiprocessing.py b/tools/test_coco_multiprocessing.py index ba91ae9..997ba08 100644 --- a/tools/test_coco_multiprocessing.py +++ b/tools/test_coco_multiprocessing.py @@ -95,7 +95,7 @@ def test_coco(det_net, real_test_img_list, eval_data, gpu_ids): start = i * nr_image end = min(start + nr_image, nr_records) split_records = real_test_img_list[start:end] - proc = Process(target=worker, args=(i, split_records, det_net, eval_data, result_queue)) + proc = Process(target=worker, args=(int(gpu_ids.strip().split(',')[i]), split_records, det_net, eval_data, result_queue)) print('process:%d, start:%d, end:%d' % (i, start, end)) proc.start() procs.append(proc)