- 博客(3)
- 资源 (1)
- 收藏
- 关注
原创 图像RGBA转BGRA方法(SIMD+TBB/cv::parallel_for_)的测试对比
RGBA2BGRA图像通道转换的几种并行优化方法以及和Opencv 的 cvtColor的对比。
2022-08-09 20:52:10 977 1
原创 Pytorch1.4下运行 GCNv2slam 需要修改的地方
1 .编译GCNextractor.h 第99行:std::shared_ptr<torch::jit::script::Module> module;改为: torch::jit::script::Module module;GCNextractor.cc 第270行:auto output = module->forward(inputs).toTuple();改为:auto output = module.forward(inputs).toTuple();
2020-05-13 08:55:21 5633 13
原创 pytorch1.4 导出模型问题记录
在pytorch1.4,使用torch.jit.trace方法导出YOLO3模型时,执行traced_script_module.save("yolov3.pt")保存后,在C++环境下进行载入测试,载入的模型参数都正确,不知什么原因输出却不对。最后将尝试将保存模型名称后缀改为.pth后traced_script_module.save("yolov3.pth"),输出正常。...
2020-03-06 17:04:02 1730 18
分享Yolov3+Libtorch测试.zip
2020-08-18
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人