HR-VITON icon indicating copy to clipboard operation
HR-VITON copied to clipboard

detail ???How to modify??

Open StartHua opened this issue 1 year ago • 3 comments

1706105970087

StartHua avatar Jan 24 '24 14:01 StartHua

StableVITON

StartHua avatar Jan 24 '24 14:01 StartHua

hi bro can you help me to fix the issue Namespace(gpu_ids='0', workers=4, batch_size=1, fp16=False, cuda='True', test_name='test', dataroot='.././kaggle/working/data/', datamode='test', data_list='pairs7.txt', output_dir='.././kaggle/working/output/', datasetting='unpaired', fine_width=768, fine_height=1024, tensorboard_dir='./data/zalando-hd-resize/tensorboard', checkpoint_dir='checkpoints', tocg_checkpoint='./eval_models/weights/v0.1/mtviton.pth', gen_checkpoint='./eval_models/weights/v0.1/gen.pth', tensorboard_count=100, shuffle=False, semantic_nc=13, output_nc=13, gen_semantic_nc=7, warp_feature='T1', out_layer='relu', clothmask_composition='warp_grad', upsample='bilinear', occlusion=True, norm_G='spectralaliasinstance', ngf=64, init_type='xavier', init_variance=0.02, num_upsampling_layers='most') Start to test %s! /usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py:558: UserWarning: This DataLoader will create 4 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary. warnings.warn(_create_warning_msg( Network [SPADEGenerator] was created. Total number of parameters: 100.5 million. To see the architecture, do print(network). Error occurred: Input and output must have the same number of spatial dimensions, but got input with spatial dimensions of [1024, 768, 3] and output size of (256, 192). Please provide input tensor in (N, C, d1, d2, ...,dK) format and output size in (o1, o2, ...,oK) format. Traceback (most recent call last): File "/content/HR-VITON/test_generator.py", line 282, in main() File "/content/HR-VITON/test_generator.py", line 276, in main test(opt, test_loader, tocg, generator) File "/content/HR-VITON/test_generator.py", line 159, in test input1 = torch.cat([clothes_down, pre_clothes_mask_down], 1) UnboundLocalError: local variable 'clothes_down' referenced before assignment

I used own 768*1024 jpg images but it show above error can you guide me how to use own image and also own image what type of metadata use like [RGB vs BGR]

Manish989927 avatar Apr 05 '24 12:04 Manish989927

Error occurred: Input and output must have the same number of spatial dimensions, but got input with spatial dimensions of [1024, 768, 3] and output size of (256, 192). Please provide input tensor in (N, C, d1, d2, ...,dK) format and output size in (o1, o2, ...,oK) format.

Manish989927 avatar Apr 05 '24 12:04 Manish989927