【记录】DeepSnake复现记录

DeepSnake复现记录

---

DeepSnake复现记录

1.具体配置：

显卡3060，cuda版本11.1，wsl，torch 1.8
（本来安装的是cuda10，所有文件都可以编译，但是demo的时候会卡住。搜索结果应该是，30系显卡只支持cuda11以上的版本）

2.关于dcnv2编译

在cuda11我是没编译成功= =，并且使用github上改进的dcnv2依然没成功
（在cuda10 torch1.1下可以编译，但是无法使用，30卡）
但是在e2ec项目中看到了解决方案

Maybe you will encounter some build errors. You can choose a plan :

You can look for another implementation of DCN-V2 and compiled successfully.
You can set cfg.model.use_dcn as False. This may result in a slight drop in accuracy.
You can install mmcv, and replace 352 line of network/backbone/dla.py as from mmcv.ops import ModulatedDeformConv2dPack as DCN, replace the deformable_groups in 353 line as deform_groups.

（1）.不使用dcnv2，使用普通的conv（链接）

精度会受到影响，并且无法使用预训练权重进行测试，需要自己重新训练。

（2）.使用mmcv里的dcnv2代替

可以测试，但是mmcv的实现略有不同需要修改输出（链接）。可以使用预训练权重进行测试。

3.训练自己的数据集

注意修改yaml里面的一些参数，比如ct_hm应该改为自己数据集的类别数

heads: {'ct_hm': 3, 'wh': 2, 'reg': 2}