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不行)

你可能感兴趣的:(pytorch,人工智能,python)