MiniBullLab / easy_ai

3 stars 1 forks source link

运行segnet报错 #151

Closed foww-0001 closed 2 years ago

foww-0001 commented 3 years ago

develop分支,通过easy_tools运行segnet报错:

./easy_tools/train_scripts/SegNet.sh /home/edge/data/VOCdevkit/car_scratch_segment/ImageSets/train.txt /home/edge/data/VOCdevkit/car_scratch_segment/ImageSets/val.txt

报错如下:

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 "/home/wfw/workspace/Test/easy_ai/easy_tools/easy_ai.py", line 12, in <module>
    from easy_tools.model_train.ai_train import EasyAiModelTrain
  File "/home/wfw/workspace/Test/easy_ai/easy_tools/model_train/ai_train.py", line 8, in <module>
    from easyai.tools.utility.copy_image import CopyImage
  File "/home/wfw/workspace/Test/easy_ai/easyai/tools/__init__.py", line 1, in <module>
    from . import offline_test
  File "/home/wfw/workspace/Test/easy_ai/easyai/tools/offline_test/__init__.py", line 1, in <module>
    from . import offline_classify_evaluation
  File "/home/wfw/workspace/Test/easy_ai/easyai/tools/offline_test/offline_classify_evaluation.py", line 6, in <module>
    from easyai.tools.offline_test.base_offline_evaluation import BaseOfflineEvaluation
  File "/home/wfw/workspace/Test/easy_ai/easyai/tools/offline_test/base_offline_evaluation.py", line 6, in <module>
    from easyai.evaluation.utility.evaluation_factory import EvaluationFactory
  File "/home/wfw/workspace/Test/easy_ai/easyai/evaluation/__init__.py", line 2, in <module>
    from . import det2d
  File "/home/wfw/workspace/Test/easy_ai/easyai/evaluation/det2d/__init__.py", line 1, in <module>
    from . import detection_mAP
  File "/home/wfw/workspace/Test/easy_ai/easyai/evaluation/det2d/detection_mAP.py", line 10, in <module>
    from easyai.data_loader.det2d.det2d_sample import DetectionSample
  File "/home/wfw/workspace/Test/easy_ai/easyai/data_loader/__init__.py", line 1, in <module>
    from . import common
  File "/home/wfw/workspace/Test/easy_ai/easyai/data_loader/common/__init__.py", line 1, in <module>
    from . import common_dataloader
  File "/home/wfw/workspace/Test/easy_ai/easyai/data_loader/common/common_dataloader.py", line 11, in <module>
    REGISTERED_TRAIN_DATALOADER.add_module(data.DataLoader, DataloaderName.DataLoader)
AttributeError: module 'torch.utils.data' has no attribute 'DataLoader'
Failed to start easy_ai
lpj0822 commented 3 years ago

我这边没有问题

lpj0822 commented 3 years ago

torch没有安装?