weixin_45650404的博客

RGBA2BGRA图像通道转换的几种并行优化方法以及和Opencv 的 cvtColor的对比。

在pytorch1.4，使用torch.jit.trace方法导出YOLO3模型时，执行traced_script_module.save("yolov3.pt")保存后，在C++环境下进行载入测试，载入的模型参数都正确，不知什么原因输出却不对。 最后将尝试将保存模型名称后缀改为.pth后traced_script_module.save("yolov3.pth"),输出正常。 ...

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();