Search code examples
python-3.xtensorflowobject-detectionobject-detection-api

How to edit the new object-detection-tutorial.ipynb


i want to edit the object_detection_tutorial.ipynb to use my trained model but the tutorial that i followed online had the old tensorflow object detection api

i tried using the old object_detection_tutorial.ipynb and continue to follow the tutorial but it didn't show bounding boxes and but when i try the original code where they detect the dog and beach it worked fine using the new object_detection_tutorial.ipynb so decided to just edit the new one but i cant really do it correctly

this is what i was saying the new object_detection_tutorial.ipynb https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb and i was following sentdex tutorial which is 2 years ago and alot has changed on the object_detection_tutorial.ipynb that was available at that time

what do i need to change in object_detection_tutorial.ipynb to be able to use my own trained model


Solution

  • You should try to just import your model in this, in the api indicated below -

    detection_model = load_model(model_name)

    this has worked for me, hope you have loaded your map file.