ZPdesu / SEAN

SEAN: Image Synthesis with Semantic Region-Adaptive Normalization (CVPR 2020, Oral)
https://zpdesu.github.io/SEAN/
Other
652 stars 95 forks source link

run_UI.py wont work #17

Closed Carebear80 closed 4 years ago

Carebear80 commented 4 years ago

sudo python run_UI.py --name CelebA-HQ_pretrained --lo/labels --image_dir datasets/CelebA-HQ/test/images --label_nc 19 --no_instance --gpu

----------------- Options --------------- aspect_ratio: 1.0
batchSize: 1
cache_filelist_read: False
cache_filelist_write: False
checkpoints_dir: ./checkpoints
contain_dontcare_label: False
crop_size: 256
dataroot: ./datasets/cityscapes/
dataset_mode: custom [default: coco] display_winsize: 256
gpu_ids: 0
how_many: inf
image_dir: datasets/CelebA-HQ/test/images [default: None] init_type: xavier
init_variance: 0.02
instance_dir:
isTrain: False [default: None] label_dir: datasets/CelebA-HQ/test/labels [default: None] label_nc: 19 [default: 13] load_from_opt_file: False
load_size: 256
max_dataset_size: 9223372036854775807
model: pix2pix
nThreads: 28
name: CelebA-HQ_pretrained [default: label2coco nef: 16
netG: spade
ngf: 64
no_flip: True
no_instance: True [default: False] no_pairing_check: False
norm_D: spectralinstance
norm_E: spectralinstance
norm_G: spectralspadesyncbatch3x3
num_upsampling_layers: normal
output_nc: 3
phase: test
preprocess_mode: resize_and_crop
results_dir: ./results/
serial_batches: True
status: test
use_vae: False
which_epoch: latest
z_dim: 256
----------------- End ------------------- QStandardPaths: XDG_RUNTIME_DIR not set, defaulting to '/tmp/runtime-root' Network [SPADEGenerator] was created. Total number of parameters: 266.9 million. To /home/user/anaconda3/envs/sean/lib/python3.6/site-packages/torch/nn/functional.py:1350: UserWarning: nn.functional.sigmoid is deprecated. Use torch.sigmoid instead. warnings.warn("nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.") /home/user/anaconda3/envs/sean/lib/python3.6/site-packages/torch/nn/functional.py:1339: UserWarning: nn.functional.tanh is deprecated. Use torch.tanh instead. warnings.warn("nn.functional.tanh is deprecated. Use torch.tanh instead.") Traceback (most recent call last): File "run_UI.py", line 566, in ex = ExWindow(opt) File "run_UI.py", line 35, in init self.EX = Ex(opt) File "run_UI.py", line 74, in init self.init_screen() File "run_UI.py", line 106, in init_screen self.run_deep_model() File "run_UI.py", line 127, in run_deep_model qim = QImage(generated_img.data, generated_img.shape[1], generated_img.shape[0], QImage.Format_RGB888) TypeError: arguments did not match any overloaded call: QImage(): too many arguments QImage(int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(bytes, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(sip.voidptr, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(bytes, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(sip.voidptr, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(List[str]): argument 1 has unexpected type 'memoryview' QImage(str, format: str = None): argument 1 has unexpected type 'memoryview' QImage(QImage): argument 1 has unexpected type 'memoryview' QImage(Any): too many arguments

I can't get the UI to work. Any ideas?

apchenstu commented 4 years ago

generated_img.data -> generated_img.data.tobytes()

Carebear80 commented 4 years ago

Omg yes!! that totally work! Thank you very much!!!