search

bubbliiiing / faster-rcnn-tf2

这是一个faster-rcnn的tensorflow2实现的库，可以利用voc数据集格式的数据进行训练。
MIT License
172 stars 50 forks source link

请问作者这个代码可以跑在CPU上吗？ #6

Open pioneer-bit opened 3 years ago

pioneer-bit commented 3 years ago

如果可以需要修改哪里的代码呢？

bubbliiiing commented 3 years ago

可以，装好tensorflow-cpu就行了