search

open-mmlab / mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.
https://mmgeneration.readthedocs.io/en/latest/
Apache License 2.0
1.9k stars 229 forks source link

CycleGAN #129

Closed guo0522 closed 2 years ago

guo0522 commented 3 years ago

I tried to inference CycleGAN by horse2zebra id0 config and weight, but got this error. Could I know if I have somethig mistake or missing ?

config_file = '/mmlab/mmgeneration/configs/cyclegan/cyclegan_lsgan_id0_resnet_in_1x1_270k_horse2zebra.py' checkpoint_file = '/mmlab/mmgeneration/cyclegan_lsgan_id0_resnet_in_1x1_266800_horse2zebra_convert-bgr_20210902_165724-77c9c806.pth' device = 'cuda:0' model = init_model(config_file, checkpoint_file, device=device)

TypeError Traceback (most recent call last) ~/miniconda3/envs/open-mmlab/lib/python3.7/site-packages/mmcv/utils/registry.py in build_from_cfg(cfg, registry, default_args) 51 try: ---> 52 return obj_cls(**args) 53 except Exception as e:

TypeError: init() got an unexpected keyword argument 'default_domain'

During handling of the above exception, another exception occurred:

TypeError Traceback (most recent call last)

in 7 device = 'cuda:0' 8 # init a generatvie ----> 9 model = init_model(config_file, checkpoint_file, device=device) 10 # translate a single image 11 translated_image = sample_img2img_model(model, image_path) ~/mmlab/mmgeneration/mmgen/apis/inference.py in init_model(config, checkpoint, device, cfg_options) 33 config.merge_from_dict(cfg_options) 34 ---> 35 model = build_model( 36 config.model, train_cfg=config.train_cfg, test_cfg=config.test_cfg) 37 ~/mmlab/mmgeneration/mmgen/models/builder.py in build_model(cfg, train_cfg, test_cfg) 29 30 def build_model(cfg, train_cfg=None, test_cfg=None): ---> 31 """Build model (GAN).""" 32 return build(cfg, MODELS, dict(train_cfg=train_cfg, test_cfg=test_cfg)) 33 ~/mmlab/mmgeneration/mmgen/models/builder.py in build(cfg, registry, default_args) 24 ] 25 return nn.ModuleList(modules) ---> 26 27 return build_from_cfg(cfg, registry, default_args) 28 ~/miniconda3/envs/open-mmlab/lib/python3.7/site-packages/mmcv/utils/registry.py in build_from_cfg(cfg, registry, default_args) 53 except Exception as e: 54 # Normal TypeError does not print class name. ---> 55 raise type(e)(f'{obj_cls.__name__}: {e}') 56 57 TypeError: CycleGAN: __init__() got an unexpected keyword argument 'default_domain'
plyfager commented 3 years ago

Hi @guo0522 , I tried your code and found no problem. I' m wondering you have not run the latest MMGEN.