首页 > Windows开发 > 详细

win10编译maskrcnn benchmark

时间:2019-05-10 18:04:33      阅读:660      评论:0      收藏:0      [点我收藏+]

步骤

1. 按照官网的Option1安装步骤安装

https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/INSTALL.md

注意:编译cocoApi会遇到-Wno-cpp相关的错误,直接将 cocoapi\PythonAPI\setup.py中的选项去掉即可,修改成红色的命令

ext_modules = [
    Extension(
        pycocotools._mask,
        sources=[../common/maskApi.c, pycocotools/_mask.pyx],
        include_dirs = [np.get_include(), ../common],
        #extra_compile_args=[‘-Wno-cpp‘, ‘-Wno-unused-function‘, ‘-std=c99‘],
        extra_compile_args=[‘-std=c99‘],
    )
]

2. maskrcnn的编译错误

no instance of function template "THCCeilDiv" matches

解决办法:将对应文件里面的"THCCeilDiv" 对应语句换掉,文件包括maskrcnn-benchmark\maskrcnn_benchmark\csrc\cuda目录下的ROIAlign_cuda.cu, ROIPool_cuda.cu, SigmoidFocalLoss_cuda.cu

//dim3 grid(std::min(THCCeilDiv(d_logits_size, 512L), 4096L));
  dim3 grid(std::min(((int)d_logits_size + 512 -1) / 512, 4096));

参考上面的示例替换即可

win10编译maskrcnn benchmark

原文:https://www.cnblogs.com/haiyang21/p/10845782.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!