search

Shanshan7 / UniMOT

A framework to do MOT
0 stars 0 forks source link

Fairmot的onnx转换 #8

Closed Shanshan7 closed 2 years ago

Shanshan7 commented 2 years ago

Fairmot带有dcn的结构目前没法通过onnx进行转换。

wc1229 commented 2 years ago

怎么查看fairmot模型的输入格式呢，转化 onnx需要model_last.pth的输入格式

wc1229 commented 2 years ago

yolo的onnx转换，没有检查其预测精度，目前在看怎么检查 链接: https://pan.baidu.com/s/1u4iUqhV3iG9nQqBzSZ4rIw 密码: ttot

wc1229 commented 2 years ago

新机器安装cuda了吗，DCNv2没有cuda用不了

foww-0001 commented 2 years ago

安装了，cuda10.2

wc1229 commented 2 years ago

nvidia-smi表示的是可以安装的最高版本，可以试一下nvcc -V命令，这个命令下显示的是没有安装

foww-0001 commented 2 years ago

拉取最新develop分支代码，可以运行onnx_inference. python3 detection/fairmot/fairmot_onnx_infer.py -i /home/edge/data/VOCdevkit/videos/MOT11/1.jpg -o ./fairmot_dlav0_34.onnx 目前运行结果还有问题，可以先自己调试一下。开发的话切自己的分支进行开发。

wc1229 commented 2 years ago

我把图片归一化之后，结果还是不对，可能需要你帮我看一下

Shanshan7 commented 2 years ago

你先把结果做一个表格进行比对：

                              原始pt推理 |  onnx 推理

输入到网络的图像 | 网络输出的数值| 后处理输出的数值|

wc1229 commented 2 years ago

另外四个模型的onnx推理结果效果都可以，代码上传到了dev_wc，接下来要做什么呢

Shanshan7 commented 2 years ago

代码已经合并，需要进行测试。