tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

list index out of range #362

Open adeindra94 opened 2 years ago

adeindra94 commented 2 years ago

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!