deformable_conv2d_pytorch's Introduction

Deformable_Conv2d_Pytorch

deformable_conv2d layer implemented in pytorch. and I wrote several articles on ZHIHU, you can read it for more detailed information
1.deformable变形卷积pytorch实现(第一节Custom op extension) 2.deformable变形卷积pytorch实现(第二节deformable_conv2d 实现一)
besides, I also complete an example net, here and I'm very sorry that I did not implement the swapaxis methods, so the im2col_step parameter are only allowed using value one.

ENVIRONMENT CONFIGURATION

OS: ubuntu16.04
GPU: 1 gtx1080Ti
LANGUAGE: python3.6.8 & c++11 & cuda c
DL FRAMEWORK: Pytorch 1.0.1
ANCILLARY LIB: setuptools: 36.4.0, numpy: 1.15.4
GPU API: NVIDIA CUDA 9.0 & cuDNN 7.0
COMPILE: nvcc & gcc 5.4.0

INSTALL PROCEDURE

cd "current project"
run mask.sh
tips: you need to modify the path parameters first. and all the -I and -L path in the nvcc and g++ orders need to be checked, make sure they are the correct path in your system
import deformable_conv2d_wrapper.py into your python file, and you can call the class BasicDeformableConv2D then.

deformable_conv2d_pytorch's People

Contributors

Stargazers

Watchers

deformable_conv2d_pytorch's Issues

ImportError: /home/amax/anaconda3/envs/fastfcn/lib/python3.6/site-packages/deformable_conv2d-0.0.0-py3.6-linux-x86_64.egg/deformable_conv2d_gpu.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail36_typeMetaDataInstance_preallocated_7E

编译以后 import deformable_conv2d_gpu，报错

>>> import deformable_conv2d_gpu
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: /home/amax/anaconda3/envs/fastfcn/lib/python3.6/site-packages/deformable_conv2d-0.0.0-py3.6-linux-x86_64.egg/deformable_conv2d_gpu.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail36_typeMetaDataInstance_preallocated_7E

ERROR: torch/extension.h and ATen/ATen.h

when i run make.sh,i get this error
'''
deformable_conv2d.cc:1:10: fatal error: torch/extension.h: No such file or directory
1 | #include <torch/extension.h>
| ^~~~~~~~~~~~~~~~~~~
compilation terminated.
deformable_conv2d_cuda.cu:3:10: fatal error: ATen/ATen.h: No such file or directory
3 | #include <ATen/ATen.h>
'''
my environment list:
python 3.6
pytorch 1.7.1
cuda 11.2
cuda_tookit 11.1
thanks a lot

Recommend Projects