DCNv2 icon indicating copy to clipboard operation
DCNv2 copied to clipboard

3D Deformable Conv.....

Open JarveeLee opened this issue 6 years ago • 5 comments

Could you please write a 3D version of deformable convolution ... I am dealing with medical data which require 3D conv but my modified 3D deformable conv from your code can not pass gradient check if threshold is small.

JarveeLee avatar Jan 09 '19 07:01 JarveeLee

my main modification is the bp for d-dimension like here,now my offset grad error is around 1e-3... grad img , grad mask , grad bias has been lower than 1e-5.... image

JarveeLee avatar Jan 10 '19 06:01 JarveeLee

But offset grad error from your 2D implementation can reach 1e-5...I don't know what is wrong here...

JarveeLee avatar Jan 10 '19 10:01 JarveeLee

@JarveeLee have you come up with an adequate 3D solution that could be shared?

jmarsil avatar Nov 01 '19 13:11 jmarsil

But offset grad error from your 2D implementation can reach 1e-5...I don't know what is wrong here...

Hi @CharlesShang @Jarveelee, I was trying to do the same thing but stuck in function dmcn_im2col_bilinear in deform_im2col_cuda.cuh, any hint about it?

Kindpire avatar Feb 01 '20 17:02 Kindpire

any update? Any one knows does the im2col implementation suitable for 3d conv? (will it be too big for memory)

hua-wu-que avatar Aug 24 '20 22:08 hua-wu-que