[bottom-up-attention] 提取object-level features问题汇总

最新推荐文章于 2024-02-16 14:52:04 发布

原创

最新推荐文章于 2024-02-16 14:52:04 发布 · 868 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #linux #caffe

在使用bottom-up-attention提取object-level features时遇到一系列问题，包括执行generate_tsv.py时的Python解释器版本错误，CUDNN_STATUS_INTERNAL_ERROR的权限问题，图像数据路径错误以及LMDB文件内存占用过大的问题。解决方案涉及修改Makefile适应Python环境，使用sudo增加权限，修正数据路径，以及通过文件映射和调整docker内存限制来缓解LMDB写入问题。

问题1
执行generate_tsv.py时，出现以下异常信息

Traceback (most recent call last):
File "./tools/generate_tsv.py", line 26, in
from fast_rcnn.test import im_detect,_get_blobs
File "/home/vilbert_beta/bottom-up-attention/tools/../lib/fast_rcnn/test.py", line 17, in
from fast_rcnn.nms_wrapper import nms, soft_nms
File "/home/vilbert_beta/bottom-up-attention/tools/../lib/fast_rcnn/nms_wrapper.py", line 9, in
from nms.gpu_nms import gpu_nms
ImportError: No module named gpu_nms