Skip to content

KeyError 1920 #9

@Kayce001

Description

@Kayce001

using the official command: python main.py --mode test --cuda 0 --version ASMAfinal --dataloader_workers 8 --testImgRoot ./bench/ --nodeName localhost --checkpoint 350000 --testScriptsName common_useage --specify_sytle 8

then error happened
Generator Script Name: Conditional_Generator_asm
11 classes
Finished preprocessing the test dataset, total image number: 25...
/home/ama/anaconda3/envs/ASMA/lib/python3.9/site-packages/torchvision/transforms/transforms.py:332: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum.
warnings.warn(
Traceback (most recent call last):
File "/home/ama/ASMAGAN/main.py", line 266, in
tester.test()
File "/home/ama/ASMAGAN/test_scripts/tester_common_useage.py", line 50, in test
test_data = TestDataset(test_img,batch_size)
File "/home/ama/ASMAGAN/data_tools/test_data_loader_resize.py", line 36, in init
transform.append(T.Resize(1088,1920))
File "/home/ama/anaconda3/envs/ASMA/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 336, in init
interpolation = _interpolation_modes_from_int(interpolation)
File "/home/ama/anaconda3/envs/ASMA/lib/python3.9/site-packages/torchvision/transforms/functional.py", line 47, in _interpolation_modes_from_int
return inverse_modes_mapping[i]
KeyError: 1920

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions