TMnet--DCN2实现过程

dcn2的编译,这个项目兼容性很差。pytorch版本不同就跑不了。

要按照这篇文章 把cuda的代码该一改即可。

https://github.com/tteepe/DCNv2.git

另外神经网络中的forward函数根构造期是一样的,在初始化的时候,他会自动调用。

tensor规格(n,c,h,w):n为样本量,bathcsize,c是通道数。输出一个x通道的的结果,则需要巻积核为c*x

conv2d里面的各个参数很显然看文档,当巻积核的大小是正方形写一个数字就好,不用(2,4)

stride也一样

我发现我现在看不懂代码的原因是:这个pytorch里面的巻积操作函数啥的不知道是干吗的

另外,forward一般都是一个网络结构定义完,会自动调用,然后forward对这个输入进行这个网络中定义的巻积操作进行

还有这个basicvsr,网络的设计就算:计算光流,然后把计算得到的光流加到这个双向传播里面去,再直接上采样。(发现把那个pcd直接用到basicsr不行)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值