logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

[目标检测]windows下实现c++版faster-rcnn

本文主要参考:http://blog.csdn.net/oYangZi12/article/details/53290426?locationNum=5&fps=11.下载微软提供的caffe(https://github.com/Microsoft/caffe)并编译,Pre-Build StepsCopy .\windows\CommonSettings.props.example

#caffe
DCNv2 pytorch1.7,1.8 cuda11.0编译

下载该工程,将DCN目录下的dcn_v2.py文件移动到上层,和make.sh文件放在同一级目录,然后执行:./make.sh。

#pytorch#深度学习
[目标检测]windows下实现c++版faster-rcnn

本文主要参考:http://blog.csdn.net/oYangZi12/article/details/53290426?locationNum=5&fps=11.下载微软提供的caffe(https://github.com/Microsoft/caffe)并编译,Pre-Build StepsCopy .\windows\CommonSettings.props.example

#caffe
swintransformer tensorrt

pytorch版本导出onnx的代码大致雷同,op12,定义input,outname,唯一注意的是最后一层argmax不要onnx导出。这个op用onnx导出直接导致推理结果全为0.如果还有提示pad的错误,可以直接修改inputsize为7的整数倍,去掉pad部分的代码。tensorrt版本本次选用8.2.3,不需要自定义plugin,layernorm ,gelu都能正确解析推理。argma

#深度学习#pytorch#人工智能
DinoV2的环境配置

其中pytorch的安装去官网按命令安装。如果运行过程中有报版本错误,可以屏蔽或者追对性修改代码。

#人工智能#深度学习#pytorch
到底了