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

你可能感兴趣的:(fasterrcnn从gpu切换到cpu版本)