这个问题是深度学习,用pytorch跑的时候出现的
解决办法:这个是格式问题,希望的格式是double,但得到的是float。字面意思是这个,但是并不是非要把格式改成double,这个时候应该在出错的前面几处代码设个断点debug一下,我得到的结果是image、img_rgb都是tensor.unit8格式的,但程序所需要的是torch.float32格式的,因此在这两行后面加上.float()即可
会出上述的错误,改成:
文章出处登录后可见!
已经登录?立即刷新
这个问题是深度学习,用pytorch跑的时候出现的
解决办法:这个是格式问题,希望的格式是double,但得到的是float。字面意思是这个,但是并不是非要把格式改成double,这个时候应该在出错的前面几处代码设个断点debug一下,我得到的结果是image、img_rgb都是tensor.unit8格式的,但程序所需要的是torch.float32格式的,因此在这两行后面加上.float()即可
会出上述的错误,改成:
文章出处登录后可见!