Skip to content

使用run_net.py with task='test' 调用Oriented_RCNN模型在FAIR1M_1_5数据运行时出现 list index out of range #56

@soaprockets

Description

@soaprockets

问题报错如下:
Traceback (most recent call last):
File "/home/hexf/data/jdet_anno/tools/run_net.py", line 57, in
main()
File "/home/hexf/data/jdet_anno/tools/run_net.py", line 52, in main
runner.test()
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jittor/init.py", line 112, in inner
ret = func(*args, **kw)
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jittor/init.py", line 280, in inner
ret = func(*args, **kw)
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jdet-0.2.0.0-py3.8.egg/jdet/runner/runner.py", line 227, in test
data_merge_result(save_file,self.work_dir,self.epoch,self.cfg.name,dataset_type,self.cfg.dataset.test.images_dir)
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jdet-0.2.0.0-py3.8.egg/jdet/data/devkits/data_merge.py", line 68, in data_merge_result
data_merge(result_pkl, save_path, final_path,dataset_type)
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jdet-0.2.0.0-py3.8.egg/jdet/data/devkits/data_merge.py", line 53, in data_merge
prepare_data(result_pkl,save_path, classes)
File "/home/hexf/.conda/envs/jdet/lib/python3.8/site-packages/jdet-0.2.0.0-py3.8.egg/jdet/data/devkits/data_merge.py", line 37, in prepare_data
classname = classes[label]
IndexError: list index out of range
task='train'时可以不会出现问题,但是单独进行test时出现问题

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions