Jittor / JDet

JDet is an object detection benchmark based on Jittor. Mainly focus on aerial image object detection (oriented object detection).
Apache License 2.0
191 stars 34 forks source link

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

Open soaprockets opened 2 years ago

soaprockets commented 2 years ago

问题报错如下: 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时出现问题