pytorch/examples

Neural Style fails if style image has an alpha channel

Open

#738 创建于 2020年3月20日

在 GitHub 查看
 (0 评论) (0 反应) (0 负责人)Python (21,634 star) (9,429 fork)batch import
help wanted

描述

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.

贡献者指南

技术栈
python
领域
machine learning
议题类型
bug
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
under 1 hour
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
fresh
清晰度议题是否清楚说明期望改动、验收标准和下一步。
clear
前置要求
PythonPIL/PillowImage handling
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
80
研究方向
The fix is to modify line 55 in fast neural style/neural style/neural style.py by appending .convert('RGB') after loading the style image. Optionally, add a warning when alpha channel is discarded. The file is part of the pytorch/examples repository. No additional research needed.