如何在 google colab tensorflow 1.15.2 上訓練自定義對象檢測模型 ssd_mobilenet_v1_coco 和 ssd_inception_v2_coco？

Question

基本上，我一直在嘗試使用 tensorflow 對象檢測 api 在 google colab tensorflow 1.15.2 上使用 ssd_mobilenet_v1_coco 和 ssd_inception_v2_coco 訓練自定義對象檢測模型。 一旦我開始訓練，它就會分別為兩個模型拋出錯誤。

我還運行了python object_detection/builders/model_builder_tf1_test.py ，它通過了所有測試，沒有任何錯誤或警告。

ValueError: ssd_inception_v2 is not supported. See ValueError: ssd_inception_v2 is not supported. See for features extractors compatible with different versions of Tensorflow. ValueError: ssd_inception_v2 is not supported. See for features extractors compatible with different versions of Tensorflow.

ValueError: ssd_mobilenet_v1_coco is not supported. See ValueError: ssd_mobilenet_v1_coco is not supported. See for features extractors compatible with different versions of Tensorflow. ValueError: ssd_mobilenet_v1_coco is not supported. See for features extractors compatible with different versions of Tensorflow.

我已經使用下面的命令成功地將 tensorflow 更改為 1.15.2，這是我安裝任何依賴項之前的第一步。

%tensorflow_version 1.x
import tensorflow
print(tensorflow.__version__)

當我檢查 model_builder.py 時，我可以看到他們仍然支持 ssd_mobilenet_v1 和 ssd_inception_v2。 我想通過將我的自定義訓練模型 ssd_mobilenet_v1 或 ssd_inception_v2 轉換為 trt-tf models 來部署在 jetson tx2 上。 在這兩個文檔https://www.elinux.org/Jetson_Zoo和https://github.com/NVIDIA-AI-IOT/tf_trt_models#od_models 中，我們可以看到可以轉換為 tf-trt 模型的對象檢測模型。 所以我的問題是如何訓練這些模型，因為它們在 tensorflow 1.15.2 上的 google colab 上得到支持，並在 jetson txt 上部署以將它們轉換為 tf-trt 模型？ 任何人都可以指導我完成它對繼續我的學習和學習一些有趣的東西真的很有幫助，謝謝

Answer 1

我認為您從 tensorflow v2 存儲庫下載了這兩個模型，而 tensorflow 1.15 顯然不支持它們。 從這里下載模型： https : //github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md然后再試一次。祝你好運