Ha0Tang / SelectionGAN

[CVPR 2019 Oral] Multi-Channel Attention Selection GAN with Cascaded Semantic Guidance for Cross-View Image Translation
http://disi.unitn.it/~hao.tang/project/SelectionGAN.html
459 stars 63 forks source link

hellow! #2

Closed HaoDing123 closed 5 years ago

HaoDing123 commented 5 years ago

Hi! when I run the train,py ,I meet a error :ModuleNotFoundError: No module named 'models.pix2pix_model' . Could you help me ?

Ha0Tang commented 5 years ago

What command did you use to run it?

HaoDing123 commented 5 years ago

Thank you for your reply! I download the dateset-dayton,and use the command:python train.py --dataroot ./datasets/dayton --name sgan --model selectiongan --which_model_netG unet_256 --which_direction AtoB --dataset_mode aligned --norm batch --gpu_ids 0 --batchSize 16 --niter 50 --niter_decay 50 --loadSize 72 --fineSize 64 --no_flip --lambda_L1 100 --lambda_L1_seg 1 --display_winsize 64 --display_id 0. Result:AssertionError: ./datasets/dayton/train is not a valid directory

Ha0Tang commented 5 years ago

Try this one: python train.py --dataroot ./datasets/dayton --name sgan --model selectiongan --which_model_netG unet_256 --which_direction AtoB --dataset_mode aligned --norm batch --gpu_ids 0 --batchSize 4 --loadSize 286 --fineSize 256 --no_flip --lambda_L1 100 --lambda_L1_seg 1 --display_id 0

and double-check your folder ./datasets/dayton.

HaoDing123 commented 5 years ago

I have tried it,but come across the same error. The folder ./datasets/dayton is full of images which is downloaded from http://www.mediafire.com/file/k01laql4bz5r7x5/dayton.zip/file. I also think it's a problem with this folder. Have you done anything else with this folder?

Ha0Tang commented 5 years ago

Can you run this first: python train.py --dataroot ./datasets/samples/dayton/ --name sgan --model selectiongan --which_model_netG unet_256 --which_direction AtoB --dataset_mode aligned --norm batch --gpu_ids 0 --batchSize 1 --loadSize 286 --fineSize 256 --no_flip --lambda_L1 100 --lambda_L1_seg 1 --display_id 0 let me know if this ok?

HaoDing123 commented 5 years ago

Hello,it doesn't work. There is no folder under dayton. it only have a lot of images.

Ha0Tang commented 5 years ago

It works now, can you try again?

HaoDing123 commented 5 years ago

It' ok! Thank you very much!