a819411321的博客

Pytorch到libtorch存在BN层参数裁剪 楼主有两个模型，两个模型都是AE。一个是用非全零图片训练的AE，一个是用大部分全零图片训练的。非全零图片训练的模样，pytorch推导结果和libtorch结果一致。大部分全零图片的AE模型的libtorch结果和pytoch结果不一致。当去除AE网络的BN层后，发现结果又一致了。猜测torch.jit.trace会裁剪bn参数。 ...

pytorch代码参考githubhttps://github.com/eriklindernoren/PyTorch-YOLOv3 部署方法使用libtorch1.0+opencv3 不建议使用libtorch1.1 另外libtorch 本人试过cpu版本和cuda9.0的gpu的版本 在windows和linux下均跑通过 linux下的编译文件写法 cmake_minimum_r...

阶段总结 libtorch实现yolov3模型部署 libtorch只使用了官方的编译 具体方法：等有时间在慢慢详细写 今天先写需要环境。 win10，vs2015，torch.1.0 opencv2.4 libtorch1.0 先上效果 ...

首先感谢https://zhuanlan.zhihu.com/p/52806730 搞了一下午终于在win10下的vs2015实现了简单的libtorch功能，接下来看看模型怎么部署。 说一下我实现的过程。 说一下我电脑的所有环境，win10 （64位）cud9.0 cudnn7.1 按上面配置去官网下载torch1.0和libtorch torch不说 说一下libtorch的怎么在...

进入某视觉公司，领了笔记本，电脑显卡是MX150。安装了显卡驱动390.87，以及cuda9和cudnn7.1。系统ubuntu16.04，nvidia-smi有显示，nvcc -V也有显示（成功显示9.0版本） 。然后安装tensorflow-gpu==1.8，成功后import 报错。。。。。。安装pytorch，import报错。无奈都装了cpu版本。（两天任务） 深深怀疑MX150不能做...