一二三四五 上山打老虎

随笔分类 -  计算机视觉

使用Pytorch的resnet34对于单个照片测试(batchsize=1)时结果不正确踩的坑
摘要：问题 测试的时候，batchsize=1时结果不正确，而batchsize>1时结果都是正确的。 设置batchsize>1,然后求 net(data[0].unsqueeze(0).to(devices[0]))的结果与单个照片测试的结果相同。 坑点 没有在测试前加上net.eval()，作用是不 阅读全文
posted @ 2022-01-01 16:15 abestxun 阅读(439) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示