fasterrcnn从gpu切换到cpu版本
2019-04-18 本文已影响0人
conner是位好少年
从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了。