search

JimmyChame / SADNet

Pytorch code for "Spatial-Adaptive Network for Single Image Denoising"
128 stars 12 forks source link

ImportError: cannot import name 'deform_conv_cuda' #1

Closed songjiechong closed 4 years ago

songjiechong commented 4 years ago

Traceback (most recent call last): File "test.py", line 10, in from model.init import make_model File "/userhome/sjc/SADNet/model/init.py", line 1, in from model.sadnet import SADNET File "/userhome/sjc/SADNet/model/sadnet.py", line 6, in from dcn.deform_conv import ModulatedDeformConvPack2 as DCN File "/userhome/sjc/SADNet/dcn/init.py", line 1, in from .deform_conv import (DeformConv, DeformConvPack, ModulatedDeformConv, ModulatedDeformConvPack, File "/userhome/sjc/SADNet/dcn/deform_conv.py", line 10, in from . import deform_conv_cuda ImportError: cannot import name 'deform_conv_cuda'

JimmyChame commented 4 years ago

You may recompile the dcn module according to your machine and then replace deform_conv.py with our file.