win10編譯maskrcnn benchmark


步驟

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));

參考上面的示例替換即可


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM