首页 / 浏览问题 / 桌面GIS / 问题详情
机器学习对象提取模型训练报错
28EXP 2024年07月03日
During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<input>", line 1, in <module>
  File "C:\SuperMap\SuperMap iDesktopX 11i\resources\python-helpers\pydev\_pydev_bundle\pydev_umd.py", line 197, in runfile
    pydev_imports.execfile(filename, global_vars, local_vars)  # execute the script
  File "C:\SuperMap\SuperMap iDesktopX 11i\resources\python-helpers\pydev\_pydev_imps\_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "C:\Users\ht\AppData\Local\Temp\model_train_7.py", line 11, in <module>
    ImageryTrainer(train_data_path=r"D:\能源遥感\training_data\PHO\training_data_凉山盐源",config=r"C:\SuperMap\SuperMap iDesktopX 11i\resources_ml\trainer_config\object_extraction\object_ext_mmdet_mask_rcnn.sdt",epoch=int(15),batch_size=int(1),lr=1.0E-4,output_model_path=r"D:\能源遥感\training_data\PHO",output_model_name=r"saved_model_凉山盐源",backbone_name=r"r-50",backbone_weight_path=r"C:\SuperMap\SuperMap iDesktopX 11i\resources_ml\backbone\resnet50-19c8e357.pth",log_path=r"D:\能源遥感\log file\0713pho_凉山盐源",reload_model=False,pretrained_model_path=None,gpus=[0]).object_extract_train()
  File "I:\teamctiy\BuildAgent\work\test_trunk/iobjectspy/ml\vision\_trainer.py", line 139, in object_extract_train
  File "I:\teamctiy\BuildAgent\work\test_trunk/iobjectspy/ml\vision\_trainer_collector\object_extraction_train.py", line 32, in train
  File "I:\teamctiy\BuildAgent\work\test_trunk/iobjectspy/ml\vision\_trainer_collector\object_extraction_train.py", line 45, in mmdet_pytorch
  File "I:\teamctiy\BuildAgent\work\test_trunk/iobjectspy/ml\vision\_models\base_framework\_mmdet\common_train_mmdet.py", line 539, in train
  File "I:\teamctiy\BuildAgent\work\test_trunk/iobjectspy/ml\vision\_models\base_framework\_mmdet\common_train_mmdet.py", line 499, in main_train
  File "C:\SuperMap\SuperMap iDesktopX 11i\support\MiniConda\conda\lib\site-packages\mmengine\runner\runner.py", line 1701, in train
    self._train_loop = self.build_train_loop(
  File "C:\SuperMap\SuperMap iDesktopX 11i\support\MiniConda\conda\lib\site-packages\mmengine\runner\runner.py", line 1493, in build_train_loop
    loop = LOOPS.build(
  File "C:\SuperMap\SuperMap iDesktopX 11i\support\MiniConda\conda\lib\site-packages\mmengine\registry\registry.py", line 570, in build
    return self.build_func(cfg, *args, **kwargs, registry=self)
  File "C:\SuperMap\SuperMap iDesktopX 11i\support\MiniConda\conda\lib\site-packages\mmengine\registry\build_functions.py", line 144, in build_from_cfg
    raise type(e)(
ValueError: class `EpochBasedTrainLoop` in mmengine/runner/loops.py: class `SMVOCMaskDataset` in iobjectspy/ml/vision/_models/base_framework/_mmdet/data/voc_mask.py: need at least one array to concatenate

训练数据生成工具运行成功 模型算法为Mask R-CNN 训练次数15 学习率0.0001

...