RuntimeError: got 4 channels instead about wct2 HOT 3 CLOSED

@jaejun-yoo <https://github.com/jaejun-yoo> Hi, here's the results I got after changing 1.png into 1.jpg in the following 4 folders ./example/content/ ./example/content_segment/ ./example/style/ ./example/style_segment/ Test the Model !python3 transfer.py --option_unpool cat5 -a --content ./examples/content --style ./examples/style --content_segment ./examples/content_segment --style_segment ./examples/style_segment/ --output ./outputs/ --verbose --image_size 1024 Namespace(alpha=1, content='./examples/content', content_segment='./examples/content_segment', cpu=False, image_size=1024, option_unpool='cat5', output='./outputs/', style='./examples/style', style_segment='./examples/style_segment/', transfer_all=True, transfer_at_decoder=False, transfer_at_encoder=False, transfer_at_skip=False, verbose=True) 0% 0/1 [00:00<?, ?it/s]invalid file (should end with .png), 1.jpg 100% 1/1 [00:00<00:00, 2674.94it/s] P.S. All the files in your examples are .png files which work fine during my test. So what's wrong with my .png files? Sincerely, Amber — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#9?email_source=notifications&email_token=ADNAIC7TLNFSSSJQUE6VEP3P5FZ73A5CNFSM4H4EMCSKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGODY44HTQ#issuecomment-507102158>, or mute the thread <https://github.com/notifications/unsubscribe-auth/ADNAIC3NVQYW35DLYLCA6ZDP5FZ73ANCNFSM4H4EMCSA> .