GPU Tutorial For Object Detection

google-coral / tutorials

Colab/Jupyter tutorials about training TensorFlow models for Edge TPU, and other tutorials

Apache License 2.0

177 stars 39 forks source link

GPU Tutorial For Object Detection #5

Closed jbrownkramer closed 2 years ago

jbrownkramer commented 4 years ago

I feel like a common use case would be to fine tune a MobileNet with a GPU and then covert everything over to edgetpu. I have created a docker that does this if there's any interest.

laurentdebricon commented 4 years ago

I am super interested ! (i don't get why the google colab tutorials provided are CPU :( still great though !)

djalusic commented 4 years ago

Hey @jbrownkramer, could you please share the tutorial? I'm currently trying to create it myself, without success.

samsterckval commented 4 years ago

I would also be interested! @jbrownkramer did you share the docker or tutorial somewhere?

Namburger commented 3 years ago

I don't think we are pursuing an official GPU one at this time, but maybe you guys can use my unofficial one? https://github.com/Namburger/edgetpu-ssdlite-mobiledet-retrain It's fully run-able on Colab!

unsurv commented 3 years ago

Screenshot_20210112_203808

Hey, I found another way:

follow the official GPU docker tutorial
in this file change FROM tensorflow/tensorflow:1.12.0-rc2-devel to FROM tensorflow/tensorflow:1.12.0-rc2-devel-gpu
When running your docker add --gpus all to your command. It should start like this: sudo docker run --name edgetpu-detect --gpus all --rm -it --privileged -p 6006:6006 You should be able to see your NVIDIA GPU from inside the docker with nvidia-smi
That's it. Follow the tutorial from coral for retraining.

hjonnala commented 2 years ago

Please check this colab GPU Tutorial for object detection: https://colab.sandbox.google.com/github/google-coral/tutorials/blob/master/retrain_efficientdet_model_maker_tf2.ipynb

google-coral-bot[bot] commented 2 years ago

Are you satisfied with the resolution of your issue? Yes No