《临时笔记》 编程生涯中的傻傻的bug问题
为了纪念在编程生涯中出现的傻傻的bug,特此做此笔记,以兹参考。
- torchvision中的ToPILImage()函数
torchvision.transforms.TOPILImage()函数用于将tensor形式的图片转化为用PIL图片格式,以便于后续的处理,比如crop,resize等等。但是这个函数的输入的张量形状应该如同(Channels, Height, Width)的,千万不要把通道放到最后去了,不然会出现意想不到的错误,而且不容易发现。在pytorch中,应该所有的输入输出都是通道优先的(channel first)。

浙公网安备 33010602011771号