Skip to content

A study repo of Deformable Conv. And tried in MNIST dataset.

Notifications You must be signed in to change notification settings

YueyueBird-su/Deformable-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Deformable-CNN

官方(torchvision)源码:

https://pytorch.org/vision/stable/_modules/torchvision/ops/deform_conv.html

一种可变形卷积网络模块,与Conv2D一样,但是多了一半参数。常用于检测形状易变的物体,在目标检测当中使用的很多。

Deformable Conv

在MNIST数据集中使用

  1. 使用pytorch框架完善lighting Data Module

  2. 分别使用常规2D 卷积和 可变形卷积 完善模型

  3. 对模型训练相同的次数

    image20220310105355599

    通过对比可以发现,可变形卷积(蓝色)在简单的模型中较优于普通卷积。

  4. 对数据采用旋转或翻转的变换作为测试集

  5. 对比两种模型对新的数据的识别程度

image20220310105648791

在对测试数据进行旋转和翻转进行测试之后,通过对比可以发现仍然是可变形卷积的精度要优于普通卷积,说明可变形卷积有一定的抗干扰能力。

About

A study repo of Deformable Conv. And tried in MNIST dataset.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages