首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何在google colab tensorflow 1.15.2上训练自定义对象检测模型ssd_mobilenet_v1_coco和ssd_inception_v2_coco?

如何在google colab tensorflow 1.15.2上训练自定义对象检测模型ssd_mobilenet_v1_coco和ssd_inception_v2_coco?
EN

Stack Overflow用户
提问于 2020-11-17 12:29:37
回答 1查看 994关注 0票数 0

基本上,我一直在尝试使用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 model_builder.py for features extractors compatible with different versions of Tensorflow.

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

通过使用下面的命令,我已经成功地将tensorflow更改为1.15.2这是我在安装任何依赖项之前的第一步。

代码语言:javascript
运行
复制
%tensorflow_version 1.x
import tensorflow
print(tensorflow.__version__)

当我检查model_builder.py时,我可以看到他们仍然支持ssd_mobilenet_v1和ssd_inception_v2。我想要在jetson tx2上部署我的自定义训练模型ssd_mobilenet_v1或ssd_inception_v2,方法是将它们转换为trt-tf模型。在这两个文档https://www.elinux.org/Jetson_Zoohttps://github.com/NVIDIA-AI-IOT/tf_trt_models#od_models中,我们可以看到可以转换为tf-trt模型的目标检测模型。所以我的问题是,如何训练这些模型,因为它们在tensorflow 1.15.2上的google colab上受支持,并在jetson txt上部署,以便将它们转换为tf-trt模型?有人能指导我吗?如果能继续我的学习并学到一些有趣的东西,那将是非常有帮助的,谢谢

EN

回答 1

Stack Overflow用户

发布于 2020-11-22 20:34:11

我认为您从tensorflow v2存储库下载了这两个模型,而tensorflow 1.15显然不支持它们。从此处下载模型:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md并试用again.good luck

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/64869378

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档