Distributed strategy Object detection

armaanpriyadarshan / Training-a-Custom-TensorFlow-2.X-Object-Detector

Learn how to train a TensorFlow Custom Object Detector with TensorFlow-GPU

83 stars 47 forks source link

Distributed strategy Object detection #16

Open francisco64 opened 3 years ago

francisco64 commented 3 years ago

Hi @armaanpriyadarshan, this tutorial has been really helpful and worked perfectly. However, I am trying to get it working on google cloud AI platform with a distributed strategy, (Multiple GPUs or TPUs). Do you know what changes I have to make to the code to do so? Thanks for your kind help

armaanpriyadarshan commented 3 years ago

Hi! I'm not too familiar with Google's Cloud Platform, although I'm pretty sure they offer TensorFlow Enterprise. I can't exactly say I have a clear idea on what to do, but I found a google article on something that might help. https://cloud.google.com/solutions/creating-object-detection-application-tensorflow. Instead of using a pre-trained model, I think substituting the custom model might work. If you want an object detection script compatible with Flask, let me know.