search

jkli1998 / DRM

Code for paper 'Leveraging Predicate and Triplet Learning for Scene Graph Generation'. (CVPR 2024)
MIT License
25 stars 0 forks source link

configs #2

Open Maritime-Moon opened 2 months ago

Maritime-Moon commented 2 months ago

Can you tell me the difference between the config files under the configs folder?

jkli1998 commented 2 months ago

Most of them inherit from the library maskrcnn for detection and segmentation. We only added configs/DRM_e2e_relation_X_101_32_8_FPN_1x.yaml for training our model.

Maritime-Moon commented 2 months ago

When I run the relation_train_net.py, I get an error： E:\anaconda\envs\DRM\python.exe E:\learn_code\DRM-main\tools\relation_train_net.py Traceback (most recent call last): File "E:\learn_code\DRM-main\tools\relation_train_net.py", line 22, in from maskrcnn_benchmark.data import make_data_loader File "E:\learn_code\DRM-main\maskrcnn_benchmark\data__init.py", line 2, in from .build import make_data_loader, get_dataset_statistics, make_data_infer_loader File "E:\learn_code\DRM-main\maskrcnn_benchmark\data\build.py", line 13, in from maskrcnn_benchmark.utils.miscellaneous import save_labels File "E:\learn_code\DRM-main\maskrcnn_benchmark\utils\miscellaneous.py", line 10, in from maskrcnn_benchmark.structures.boxlist_ops import boxlist_iou File "E:\learn_code\DRM-main\maskrcnn_benchmark\structures\boxlist_ops.py", line 7, in from maskrcnn_benchmark.layers import nms as _box_nms File "E:\learn_code\DRM-main\maskrcnn_benchmark\layers\init.py", line 10, in from .nms import nms File "E:\learn_code\DRM-main\maskrcnn_benchmark\layers\nms.py", line 8, in nms = amp.float_function(_C.nms) File "E:\anaconda\envs\DRM\lib\site-packages\yacs\config.py", line 141, in getattr__ raise AttributeError(name) AttributeError: nms I looked and found that I didn't find _C.nms

Maritime-Moon commented 2 months ago

Can you provide information about _C.nms?

jkli1998 commented 2 months ago

This error appears to be caused by compiling this repo uncorrectly, you need to carefully follow INSTALL.md to install this code step by step, especially the last command python setup.py build develop.