吐血总结:windows下libtorch调用pytorch模型,并读取本地图片进行预测
如何保证python中torch模型输出的数据和libtorch的数据一模一样?网上各种教程,全不说关键所在。吐血三天整出了代码核心!!!怀着鸡冻的心情写下这个博客,做个记录。网上有出多使用这种torch.ones(1,3,64,64)来生成一个张量作为模型的输入数据,比较python和c++平台的输出结果是否一致。这个也是有必要的,但这只是第一步。因为我们一般输入的数据不可能都是这样生成,往往都是读取图片,然后经过一系列处理才送进module。在这个过程就会出现各种问题。这就是博主今天要总结的东
原创
2020-09-26 15:43:16 ·
3384 阅读 ·
1 评论