I'm looking into training an object detection network using Tensorflow, and I had a look at the TF2 Model Zoo. I noticed that there are noticeably less models there than in the directory /models/research/models/
, including the MobileDet with SSDLite developed for the jetson xavier.
To clarify, the readme says that there is a MobileDet GPU with SSDLite, and that the model and checkpoints trained on COCO are provided, yet I couldn't find them anywhere in the repo.
How is one supposed to use those models?
I already have a custom-trained MobileDetv3 for image classification, and I was hoping to see a way to turn the network into an object detection network, in accordance with the MobileDetv3 paper. If this is not straightforward, training one network from scratch could be ok too, I just need to know where to even start from.
1条答案
按热度按时间jw5wzhpr1#
如果您计划使用对象检测API,则不能使用现有模型。您必须从v2的here和v1的x1e1 f1 a模型列表中进行选择
文档维护得很好,TensorFlow团队对自定义数据的训练、验证或推理(测试)步骤进行了很好的here解释。该链接适用于TensorFlow v2版。但是,如果您希望使用v1,该过程相当相似,并且有许多博客/视频解释如何使用