我正在尝试构建一个使用对象检测的应用程序。到目前为止,我使用了tensorflow github中的示例:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android
他们已经在该示例中提供了一些预训练模型。事实证明,其中一个模型在我的应用程序的延迟和准确性方面是高效的。更具体地说,我指的是以下模型" ssd_mobilenet_v1_android_export.pb"在assets文件夹中,似乎他们的模型是使用COCO数据集训练的,大约有100个类。
但是我想扩展探测器以检测一些额外的类,所以我想训练新的模型。为此,我想使用darknet,但为了做到这一点,我需要.cfg和.weights文件。但是我尝试在网上搜索它,但是我无法找到使用哪个.cfg和.weights文件来生成ssd_mobilenet_v1_android_export.pb"文件。
提前谢谢。