AttributeError: module 'lib.fpn.roi_align._ext.roi_align' has no attribute 'roi_align_forward_cuda'

yuweihao / KERN

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

MIT License

120 stars 35 forks source link

AttributeError: module 'lib.fpn.roi_align._ext.roi_align' has no attribute 'roi_align_forward_cuda' #26

Open Waiting-TT opened 2 years ago

Waiting-TT commented 2 years ago

您好，我在运行“bash ./scripts/train_kern_predcls.sh”时报错如下： AttributeError: module 'lib.fpn.roi_align._ext.roi_align' has no attribute 'roi_align_forward_cuda'

请问能否解答一下，确实没有看到定义roi_align_forward_cuda，谢谢！

yuweihao commented 2 years ago

Hi @Waiting-TT , make sure to install the environment described in the README. I have just merged a PR from AU-Nebula that may help solve the difficulties of environment installation.