Reduce number of objects a pretrained Tensorflow model detects

混江龙づ霸主 提交于 2019-12-25 01:37:31

问题


I am using this code for object detection and it outputs 100 boxes even though in most pictures there are 0-5 objects. The detection takes 5 seconds on a 250X250 image. Would cutting the number of objects to be detected speed up the process, and if yes is there a way to do it?


回答1:


Logically it would, i cant say exactly by how much. You can retrain the model to only fewer objects that you are interested in. For training you will also specify objects interested in a file with extension *.pbtxt which has list of objects specified.

Few Links which talk about retraining in detail https://pythonprogramming.net/training-custom-objects-tensorflow-object-detection-api-tutorial/

https://towardsdatascience.com/building-a-toy-detector-with-tensorflow-object-detection-api-63c0fdf2ac95

https://medium.com/@dana.yu/training-a-custom-object-detection-model-41093ddc5797



来源:https://stackoverflow.com/questions/50866894/reduce-number-of-objects-a-pretrained-tensorflow-model-detects

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!