princeton-vl/CornerNet-Lite

Problems when i evulate my own data

qingzhoudong opened this issue · 0 comments

after i trained my own datasets, i want to evaluate AP, but there are some problems that i can't solve.
'''
File "evaluate.py", line 106, in
main(args)
File "evaluate.py", line 102, in main
test(testing_db, system_config, model, args)
File "evaluate.py", line 62, in test
test_func(system_config, db, nnet, result_dir, debug=debug)
File "/home/qzdong/CornerNet-Lite-master/core/test/init.py", line 5, in test_func
return globals()[sys_config.sampling_function](db, nnet, result_dir, debug=debug)
File "/home/qzdong/CornerNet-Lite-master/core/test/cornernet_saccade.py", line 282, in cornernet_saccade
detections = db.convert_to_coco(top_bboxes)
File "/home/qzdong/CornerNet-Lite-master/core/dbs/coco.py", line 143, in convert_to_coco
category_id = self._cls2coco[cls_ind]
KeyError: 6
'''
what should i do to solve this problem?