ZPdesu/SEAN

run_UI.py wont work

Carebear80 opened this issue · 2 comments

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?

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

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