【成功运行】CenterTrack代码复现及问题汇总

项目场景：

CenterTrack代码地址： CenterTrack

在 Tracking Objects as Points 上的拓展论文，模型将当前帧、前一帧以及从之前的跟踪结果中渲染的热图作为输入，并预测当前检测热图及其到前一帧中心的偏移量。主要用于跟踪层面，可以轻松拓展3D tracking、multi-category tracking、pose tracking 。

基础环境

win10 + RTX3060 + cuda11.0 + pytorch 1.7.1
其他模型所需包可以通过 cd 到项目文件夹中 运行指令 来下载：

pip install -r requirements.txt

官方安装步骤 ：安装步骤
建议先安装好pytorch，将git clone下来的 DCNv2 代码放到 CenterTrack\src\lib\model\networks 下，成功编译后在进行其他包的安装，即步骤5。具体 DCNv2 安装可以看这篇 DCNv2踩坑记录

---

问题描述

RuntimeError: CUDA error: no kernel image is available for execution on the device

RuntimeError: CUDA error: no kernel image is available for execution on the device：

---

原因分析及解决方案：

还是cuda版本问题，由于30系显卡算力太高不匹配导致，有可能你的DCNv2已经顺利编译仍会出现这个问题，比较坑，可以尝试下载高版本cuda，或者切换一个torch版本（要匹配你的cuda)。

---

问题描述：

import DCN failed
这个比较容易解决没截图

原因分析及解决方案：

检查你的DCNv2是否缺东西，有时候从github上扒代码有的新更新的子模块下载不全，建议使用命令来克隆代码，或者手动下载。

git clone --recursive

如果你的 DCNv2 成功运行后出现这个问题, 检查 DCNv2 代码文件是否放在了正确的位置： CenterTrack\src\lib\model\networks，如果你是按照我 DCNv2踩坑记录 这篇文章下载的 新版的DCNv2,记得将 dcn_v2.py 从 DCN 文件夹中移出来，放到DCNv2文件夹下。

---

问题描述

cv2.error: OpenCV(4.5.2) :-1: error: (-5:Bad argument) in function 'line '

Overload resolution failed:

• Can’t parse ‘pt1’. Sequence item with index 0 has a wrong type
• Can’t parse ‘pt1’. Sequence item with index 0 has a wrong type

原因分析及解决方案

这是由于安装了错误版本的 OpenCV。
我解决了pip install opencv-python==3.3.1.11（从 conda 环境中执行此操作）
可以参考这个连接 ：link1

---

问题描述

在采用 --save_video 之后发现 result 文件夹下没有保存已生成的视频，并且报错

OpenCV: FFMPEG: tag 0x44495658/'XVID' is not supported with codec id 13 and format 'mp4 / MP4 (MPEG-4 Part 14)'

原因分析

参考这个博主的文章参考文章 发现这是因为pip安装的opencv不支持H264格式的视频输出（版权问题）。

解决方案

尝试使用输入视频的高度和宽度指定 --video_h 和 --video_w 参数，以模型下自带视频为例，输入命令即可。

python demo.py tracking,ddd --load_model ../models/nuScenes_3Dtracking.pth --dataset nuscenes --pre_hm --track_thresh 0.1 --demo ../videos/nuscenes_mini.mp4 --test_focal_length 633 --save_video --video_h 450 --video_w 800 

最后

附上代码讨论区链接： 问题讨论区。如果你碰到了其他问题，可以去问题讨论区碰碰运气看看有没有和你相似的问题并且有人解答了。
祝大家好运，一次成功，码运新隆！！！
惯例贴个图，运行demo.

补充：

预训练权重放到网盘里了，有需要的小伙伴自己去取吧！
百度网盘链接
提取码：h0uc