MiniBullLab/easy_ai

edge_tools分支,客户训练segnet报错

Closed this issue · 1 comments

报错信息如下:

Traceback (most recent call last):
  File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/usr/local/lib/python3.6/dist-packages/easy_tools/easy_ai.py", line 67, in <module>
    train_main()
  File "/usr/local/lib/python3.6/dist-packages/easy_tools/easy_ai.py", line 60, in train_main
    train_process.segment_model_train(dir_name)
  File "easy_tools/easyai_train.py", line 59, in easy_tools.easyai_train.EasyAiModelTrain.segment_model_train
  File "easyai/train_task.py", line 29, in easyai.train_task.TrainTask.train
  File "easyai/tasks/seg/segment_train.py", line 76, in easyai.tasks.seg.segment_train.SegmentionTrain.train
  File "easyai/tasks/seg/segment_train.py", line 143, in easyai.tasks.seg.segment_train.SegmentionTrain.test
  File "easyai/tasks/seg/segment_test.py", line 45, in easyai.tasks.seg.segment_test.SegmentionTest.test
  File "easyai/evaluation/segmention_metric.py", line 19, in easyai.evaluation.segmention_metric.SegmentionMetric.numpy_eval
  File "easyai/evaluation/segmention_metric.py", line 51, in easyai.evaluation.segmention_metric.SegmentionMetric.fast_hist
IndexError: boolean index did not match indexed array along dimension 0; dimension is 200000 but corresponding boolean dimension is 9360

由于没有使用最新的标注工具,所以导致输入图片数据和标注数据的尺寸不一样,建议后面在这边加一个判断打印报错信息。
@lpj0822