《临时笔记》 编程生涯中的傻傻的bug问题

为了纪念在编程生涯中出现的傻傻的bug,特此做此笔记,以兹参考。

  1. torchvision中的ToPILImage()函数
    torchvision.transforms.TOPILImage()函数用于将tensor形式的图片转化为用PIL图片格式,以便于后续的处理,比如cropresize等等。但是这个函数的输入的张量形状应该如同(Channels, Height, Width)的,千万不要把通道放到最后去了,不然会出现意想不到的错误,而且不容易发现。在pytorch中,应该所有的输入输出都是通道优先的(channel first)。
posted @ 2020-04-27 17:49  FesianXu  阅读(51)  评论(0编辑  收藏  举报