search

ZhangHanbo / Visual-Manipulation-Relationship-Network-Pytorch

MIT License
41 stars 13 forks source link

ImportError: cannot import name 'C_ROIPooling' from 'model.roi_layers' #10

Open BHWWS opened 3 years ago

BHWWS commented 3 years ago

Traceback (most recent call last): File "main.py", line 27, in from model.utils.net_utils import weights_normal_init, save_net, load_net, \ File "/content/drive/MyDrive/VMRN-Pytorch/model/utils/net_utils.py", line 11, in from model.roi_layers import nms File "/content/drive/MyDrive/VMRN-Pytorch/model/roi_layers/init.py", line 5, in from .roi_align import ROIAlign, RoIAlignAvg, RoIAlignMax File "/content/drive/MyDrive/VMRN-Pytorch/model/roi_layers/roi_align.py", line 9, in from . import C_ROIPooling ImportError: cannot import name 'C_ROIPooling' from 'model.roi_layers' (/content/drive/MyDrive/VMRN-Pytorch/model/roi_layers/init.py)

In roi_layers/init.py, I have not find the C_ROIPooling class, so how can I solve this problem?