search

Syn-McJ / TFClassify-Unity

An example of using Tensorflow with Unity for image classification and object detection.
MIT License
167 stars 47 forks source link

[Help!!] how to use own object detection model..? #32

Closed YouMinJung closed 5 years ago

YouMinJung commented 5 years ago

I'm not familiar to tensorflow and unity... I wanna use pre-trained tensorflow model in zoo. (ssd mobilenet v2 coco) But there are 5 files in coco model. (checkpoint, frozen_inference_gragh.pb, model.ckpt.data-00000-of-00001, model.ckpt.index, model.ckpt.meta)

I wanna .pb to bytes... but I don't know how to convert it. I searched the information using google. It was told to use the freeze_gragh. I tried to use it. but I don't have .ckpt file... (I only have .ckpt.meta etc..) How can I convert the file...?

ps. Exectly, What kind of model did you uesd?

Syn-McJ commented 5 years ago

Hi @YouMinJung, I used ssd_mobilenet from TensorFlow examples.

You only need 2 files - frozen_inference_gragh.pb and file with labels. You simply have to rename the file to change .pb extension to .bytes, no need to convert anything.

Be aware that, as I mentioned in README, the performance of Detect sample isn't very good.