tusen-ai/simpledet

list index out of range

adeindra94 opened this issue · 0 comments

Hello, I had an error using the network (https://github.com/TuSimple/simpledet/blob/master/config/tridentnet_r50v1c4_c5_1x.py).

I use Pascal VOC (https://github.com/TuSimple/simpledet/blob/master/utils/create_voc_roidb.py)

and then train the model without a problem:
python3 detection_train.py --config config/tridentnet_r50v1c4_c5_1x.py

But when I do the testing:
(simpledet) doodles@cgu:/data/doodles/simpledet$ python3 detection_test.py --config config/tridentnet_r50v1c4_c5_1x.py

Traceback (most recent call last):
File "/home/doodles/miniconda3/envs/simpledet/lib/python3.7/multiprocessing/pool.py", line 121, in worker
result = (True, func(*args, **kwds))
File "/home/doodles/miniconda3/envs/simpledet/lib/python3.7/multiprocessing/pool.py", line 44, in mapstar
return list(map(*args))
File "detection_test.py", line 258, in do_nms
dataset_cid = coco.getCatIds()[cid]
IndexError: list index out of range
"""

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "detection_test.py", line 268, in
output_dict = pool.map(do_nms, output_dict.keys())
File "/home/doodles/miniconda3/envs/simpledet/lib/python3.7/multiprocessing/pool.py", line 268, in map
return self._map_async(func, iterable, mapstar, chunksize).get()
File "/home/doodles/miniconda3/envs/simpledet/lib/python3.7/multiprocessing/pool.py", line 657, in get
raise self._value
IndexError: list index out of range

Segmentation fault: 11

Segmentation fault (core dumped)

May I know how to solve this issue?

Thanks!