Ubuntu复现TraDeS

环境：Ubuntu16.04+pytorch1.3+torchvision0.4.2+python3.6+cuda10.1+cudnn7.6.5
文章链接：https://arxiv.org/pdf/2103.08808.pdf
GitHub链接：https://github.com/JialianW/TraDeS

注意事项：
1.最好重建虚拟环境，避免各种问题。原有Tracktor++环境下编译DCNv2时报错：Runtime Error: Error compiling objects for extension。查询博客，解决未果。重建环境，以GitHub为准，成功编译。
2.训练好的模型要pa qiang下载，百度网盘链接：https://pan.baidu.com/s/1Ah1E5GULE5fOTrX_-Gcc1Q 提取码：wwhh 复制这段内容后打开百度网盘手机App，操作更方便哦。
3.运行demo.py时报错：ModuleNotFoundError: No module named ‘dataset.dataset_factory’。解决办法：在dataset_factory所属文件夹内加一个__init__.py，可直接从…\TraDeS\src\lib\utils中复制，这样就能使IDE识别它为package，从而解决问题。
4.运行时报错： if pred_mask in item: 找不到pred_mask，修改为if (‘pred_mask’ in item): 运行成功。
下图为跑3D模型时的截图。

欢迎指出不足。