• DepthAI

  • MobileNetSSD training colab notebook broken

Hi,
I'm trying to train a mobileNetSSD model using the colab notebooks here - https://github.com/luxonis/depthai-ml-training/tree/master/colab-notebooks

Because of changes in colab environment from google, the installation of tensorflow 1.15 fails.

!pip install tensorflow-gpu==1.15
This has recently stopped working. I tried downgrading python in colab to 3.7. Still doesn't work. Thus, the training pipeline is broken. Can someone please help with this urgently?

Also, the training notebooks for mobilenetSSD say "Warning! This tutorial is now deprecated. Please refer to YoloV5 tutorial, YoloV6 tutorial or YoloV7 tutorial."

It is quite confusing to see a depreciation note but no new colab notebook for MobileNetSSD. Can someone clarify this as well please?

  • erik replied to this.

    Hi msharad19 ,
    We apologize for the inconvenience. Could you please create an Issue on the depthai-ml-training repo so we can track the issue?
    Thanks, Erik

      Hi Eric,
      There's already an issue for mobilenetSSD training-
      https://github.com/luxonis/depthai-ml-training/issues/32

      The comments say that since mobilenetSSD notebooks are old, they have been depreciated and users are recommended to use yolo models.

      But for us who are using mobilenetSSD in our product pipeline, it means we have to re-do a lot of work. That isn't practical.

      Has depthai decided to stop all support for mobilenetSSD training? That would lead to backward incompatibility 🙁

        7 days later

        Hi msharad19 ,
        Depthai didn't decide to stop support for mobilenetssd, but simply our ML team decided to only support yolo models, as they are opensource and more popular. You can still train and deploy mobilenetSSD model, but since colab doesn't support it you would have to train the model on your computer (instead of colab notebook). Thoughts?
        Thanks, Erik