Tensorflow Object Detection API

我们两清 提交于 2019-12-20 19:33:04

问题


I decided to take a dip into ML and with a lot of trial and error was able to create a model using TS' inception.

To take this a step further, I want to use their Object Detection API. But their input preparation instructions, references the use of Pascal VOC 2012 dataset but I want to do the training on my own dataset.

Does this mean I need to setup my datasets to either Pascal VOC or Oxford IIT format? If yes, how do I go about doing this?

If no (my instinct says this is the case), what are the alternatives of using TS object detection with my own datasets?

Side Note: I know that my trained inception model can't be used for localization because its a classifier

Edit:

For those still looking to achieve this, here is how I went about doing it.


回答1:


The training jobs in the Tensorflow Object Detection API expect to get TF Record files with certain fields populated with groundtruth data.

You can either set up your data in the same format as the Pascal VOC or Oxford-IIIT examples, or you can just directly create the TFRecord files ignoring the XML formats.

In the latter case, the create_pet_tf_record.py or create_pascal_tf_record.py scripts are likely to still be useful as a reference for which fields the API expects to see and what format they should take. Currently we do not provide a tool that creates these TFRecord files generally, so you will have to write your own.




回答2:


Except TF Object Detection API you may look at OpenCV Haar Cascades. I was starting my object detection way from that point and if provide well prepared data set it works pretty fine.

There are also many articles and tutorials about creating your own cascades, so it`s easy to start. I was using this blog, it helps me a lot.



来源:https://stackoverflow.com/questions/44810459/tensorflow-object-detection-api

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