search

see-- / keras-centernet

A Keras implementation of CenterNet with pre-trained model (unofficial)
MIT License
337 stars 84 forks source link

Can tkeras-centernet work on CPU? #6

Closed nkliuyifang closed 5 years ago

nkliuyifang commented 5 years ago

Hello,

Thanks for your work! Can this code work on CPU?

Thanks a lot~

see-- commented 5 years ago

Yes it will run on the CPU. Just install tensorflow instead of tensorflow-gpu. Anyway I wouldn't do it. It will be very slow.