fasterrcnn从gpu切换到cpu版本

从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了。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容