search

CharlesShang / TFFRCNN

FastER RCNN built on tensorflow
MIT License
875 stars 418 forks source link

CPU Mode? #81

Open KojiShiono opened 7 years ago

KojiShiono commented 7 years ago

Hello, I would like to use this in CPU-only environment, and I was following this page but still getting following error upon build. make.sh: line 9: nvcc: command not found clang: error: no such file or directory: 'roi_pooling_op.cu.o' make.sh: line 24: nvcc: command not found clang: error: no such file or directory: 'psroi_pooling_op.cu.o'

Apparently, this is how it is configured in make.sh, and I am wondering if there is any way I can build this for CPU-only environment.

Thank you in advance!

LouieYang commented 7 years ago

For roi_pooling_op, just follow this repo, which could work in this repo too. For psroi_pooling_op, I think it is not used for some network like VGGnet_test, so you could just skip compiling it and annotate some sentences in lib/networks/init.py (just leave your testing network and annotate others) and lib/networks/network.py (annotate psroi_layer).

I think there are also some lines of code that should be annotated, just follow your make error and things could be done.

Btw, it takes 12-13s to run the forwarding for my trained model on 2014 macOS CPU mode. So, I think running it on GPU is a better way.