运行ganomaly报错 - Githubissues

python3 easyai/train_task.py -t one_class -i /home/edge/data/VOCdevkit/MVtec/bottle/ImageSets/train.txt -v /home/edge/data/VOCdevkit/MVtec/bottle/ImageSets/train.txt -m GANomaly

报错如下：

2021-09-01 16:28:39,342 ERROR   [backbone_factory.py, 47] Traceback (most recent call last):
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model_block/utility/backbone_factory.py", line 38, in get_backbone_model
    result = self.get_gan_base_model(model_config)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model_block/utility/backbone_factory.py", line 83, in get_gan_base_model
    result = build_from_cfg(model_config, REGISTERED_GAN_D_BACKBONE)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/utility/registry.py", line 109, in build_from_cfg
    return obj_cls(**args)
TypeError: __init__() got an unexpected keyword argument 'pretrained'

2021-09-01 16:28:39,342 ERROR   [backbone_factory.py, 48] __init__() got an unexpected keyword argument 'pretrained'
2021-09-01 16:28:39,342 ERROR   [train_task.py, 42] Traceback (most recent call last):
  File "easyai/train_task.py", line 37, in train
    task = build_from_cfg(task_args, REGISTERED_TRAIN_TASK)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/utility/registry.py", line 109, in build_from_cfg
    return obj_cls(**args)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/tasks/one_class/one_class_train.py", line 20, in __init__
    self.set_model(gpu_id=gpu_id, init_type="normal")
  File "/home/edge/workspace/EDGE/easy_ai/easyai/tasks/utility/base_train.py", line 82, in set_model
    self.model = self.torchModelProcess.create_model(self.model_args, gpu_id)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/torch_utility/torch_model_process.py", line 29, in create_model
    model = self.modelFactory.get_model(model_config)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model/utility/model_factory.py", line 39, in get_model
    result = self.get_model_from_name(model_args)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model/utility/model_factory.py", line 71, in get_model_from_name
    model = build_from_cfg(model_config, REGISTERED_GAN_MODEL)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/utility/registry.py", line 109, in build_from_cfg
    return obj_cls(**args)
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model/gan/ganomaly.py", line 22, in __init__
    self.create_block_list()
  File "/home/edge/workspace/EDGE/easy_ai/easyai/model/gan/ganomaly.py", line 37, in create_block_list
    d_out_channels = discriminator.get_outchannel_list()
AttributeError: 'NoneType' object has no attribute 'get_outchannel_list'

2021-09-01 16:28:39,342 ERROR   [train_task.py, 43] 'NoneType' object has no attribute 'get_outchannel_list'

MiniBullLab / easy_ai

运行ganomaly报错 #142