美文网首页
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