美文网首页
fasterrcnn从gpu切换到cpu版本

fasterrcnn从gpu切换到cpu版本

作者: conner是位好少年 | 来源:发表于2019-04-18 16:45 被阅读0次

    从git拿到的程序是gpu版本的,如果想要在本地cpu执行是需要做如下修改的:

    第一步:将/lib/model/nms_wrapper.py中

    from nms.gpu_nms import gpu_nms注释掉

    第二步:将方法nms(dets, thresh, force_cpu=False)中的force_cpu的false改为true,表示强制启动cpu版本。

    第三步:将/lib下的setup.py中:

    ?xml version="1.0" encoding="UTF-8"?

    注释掉如上图所示

    然后执行make

    就ok了。

    相关文章

      网友评论

          本文标题:fasterrcnn从gpu切换到cpu版本

          本文链接:https://www.haomeiwen.com/subject/jrmtgqtx.html