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