sangyun884 / HR-VITON

Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).
825 stars 171 forks source link

please fix this issue #90

Open Manish989927 opened 4 months ago

Manish989927 commented 4 months ago

https://colab.research.google.com/drive/1BG-4G6MYmwRZlqctbVVSMC-lj27Cycoh?usp=sharing

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