Support other tensorflow object detection models

[kodonnell]

Currently one can run other models using TF_ANNOTATION_MODEL_PATH (see here) - though I guess you'd need to modify this in the dockerfile somehow (maybe here - though you'd still need to get the file onto the image).

however, the class labels are hard-coded which, in practice, means only a small subset of tensorflow object detections will work (i.e. those trained on COCO (?) which hopefully have the class names in the correct order).

Happy to submit a PR. I was thinking of doing something hacky-ish like reading the classes from a json at TF_ANNOTATION_CLASSES_PATH but then I realised that this isn't that great, as the same model would be used on all tasks i.e. you couldn't use model X for one annotation task, and model Y for another. So, I'm thinking that in the GUI, the user can upload a model (frozen graph + classes), which gets used - in the same way we upload videos etc. I feel like this should be pretty straightforward plumbing, since all of the code to run the model etc. already exists.

If that sounds like a good idea, I can have a play. Otherwise, if it's a bad idea, can someone propose an alternative? And - if there's not enough interest in this feature, then I can just hack something out for my own use case (which will be much quicker and easier for me, after all).

[nmanovic]

@kodonnell ,

We are working to implement the feature already. I hope to add it next year.

[savan77]

we have already implemented this feature for our local usage and working on adding tracking for videos. we are also planning to open-source once everything is tested thoroughly.

[nmanovic]

@savan77 ,

Awesome. Don't hesitate to send as PR. We are open for collaboration.

[kodonnell]

@savan77 - great work. Maybe if you submit a PR now, then we can also try it out and test it?

[savan77]

@kodonnell @nmanovic
I am more than happy to send a PR but I am just a developer not an owner. However, I have been told that the code will be open-sourced soon. I can't comment on exactly when.

[outan]

@kodonnell @nmanovic
I am more than happy to send a PR but I am just a developer not an owner. However, I have been told that the code will be open-sourced soon. I can't comment on exactly when.

I am very happy to wait for your PR.
I really need the tracking feature in my project to greatly reduce annotation time.
And I think that is needed in many projects.

[zishanahmed08]

@savan77 any updates?