help wanted
Repository metrics
- Stars
- (21,634 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
In fast_neural_style/neural_style/neural_style.py line 55, if the style image has an alpha channel, then the generated tensor has 4 dimensions and this causes utils.normalize_batch to throw due to a tensor dimension mismatch a few lines down.
I've fixed this by appending .convert('RGB') so line 55 now reads
style = utils.load_image(args.style_image, size=args.style_size).convert('RGB')
The ImageFolder data loader does the same transformation, however, maybe a warning should be issued since it is the key file.