DSFNet(Dynamic and Static Fusion Network for Moving Object Detection in Satellite Videos)代码调试

最新推荐文章于 2024-05-20 19:08:38 发布

爱吃糖葫芦的大熊

最新推荐文章于 2024-05-20 19:08:38 发布

阅读量1k

点赞数 2

文章标签：目标检测人工智能计算机视觉

本文链接：https://blog.csdn.net/dhdhhy/article/details/127802228

版权

帮人解决的论文代码，分享以下经验，小的安装问题就不叙述了，分析主要问题：

问题一：DCNv2的配置

运行之前还需配置好DCN，在./lib/model/DCNv2/下面因为DSFNET.py等都需要DCN网络

代码自带的DCNv2总是无法setup，报错cuda和gcc编译器问题，太麻烦，应此另辟蹊径，不使用代码自带的DCNv2，使用mmcv中的DCN实现。

首先下载安装mmcv库：

# 命令行输入：
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
# 将其中的{cu_version}替换为你的CUDA版本，{torch_version}替换为你已经安装的pytorch版本；
# 例如：CUDA 为11.0，pytorch为1.7.0
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html

刚开始一直安装不上，后面发现是信任问题，因此改为如下，将安装网站用trusted-host转换

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.13.0/index.html --trusted-host download.openmmlab.com

在代码中替换

from mmcv.ops import DeformConv2dPack as DCN

# 使用方法与官方DCNv2一样，只不过deformable_groups参数名改为deform_groups即可，例如：
dconv2 = DCN(in_channel, out_channel, kernel_size=(3, 3), stride=(2, 2), padding=1, deform_groups=2)

在原论文代码中就是将DSFNET.py、DSFNET_with_Dynamic.py、DSFNET_with_Static.py中DCN的引用做替换：

# from lib.models.DCNv2.dcn_v2 import DCN
from mmcv.ops import DeformConv2dPack as DCN

问题二：原代码的get_pose_net函数未定义

在DSFNET.py、DSFNET_with_Dynamic.py、DSFNET_with_Static.py中都提到了get_pose_net，但代码中并未定义，这其实是伪函数，需要自己定义网络如下：

在DSFNET.py中

if __name__ == '__main__':
    # net = get_pose_net({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    net = DSFNet({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    total = sum([param.nelement() for param in net.parameters()])
    print('   Number of params: %.2fM' % (total / 1e6))

在DSFNET_with_Dynamic.py中

if __name__ == '__main__':
    # net = get_pose_net({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    net = DSFNet_with_Dynamic({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    total = sum([param.nelement() for param in net.parameters()])
    print('   Number of params: %.2fM' % (total / 1e6))

在DSFNET_with_Static.py中

if __name__ == '__main__':
    # net = get_pose_net({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    net = DSFNet_with_Static({'hm': 1, 'wh': 2, 'reg': 2}, 256)
    total = sum([param.nelement() for param in net.parameters()])
    print('   Number of params: %.2fM' % (total / 1e6))