Я хочу использовать тензор потока для обнаружения автомобилей во встроенной системе, поэтому я попробовал ssd_mobilenet_v2 , и это на самом деле очень хорошо для меня, за исключением некоторых конкретных c типов автомобилей, которые не очень распространены и Я думаю, именно поэтому модель их не распознает. У меня есть набор данных этих случаев, и я хочу улучшить модель, настроив ее. Я также должен отметить, что мне нужен файл .tflite , потому что я использую tflite_runtime в python. Я следовал этим инструкциям https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10, и я смог обучить модель и достиг разумного значения потерь. Затем я использовал export_tflite_ssd_graph.py в API обнаружения объектов для построения inference_graph из обученной модели. Впоследствии я использовал инструмент toco для создания из него файла .tflite.
Но вот проблема, после того, как я все это сделал; Не только модель не улучшилась, но теперь она не обнаруживает никаких автомобилей. Я запутался и не знаю, в чем проблема, я много искал и не нашел учебника о том, что мне нужно делать. Они просто добавили новый объект в модель, а затем экспортировали его, что я попробовал, и мне это удалось. Я также попытался создать файл .tflite без обучения модели и непосредственно из модели модели обнаружения Tensorflow zoo , и она работала нормально. Поэтому я думаю, что проблема как-то связана с тренировочным процессом. Может быть, я чего-то там упускаю.
Еще одна вещь, которую я не нашел в документах, это то, можно ли «добавить» класс к текущим классам модели обнаружения объектов. Например, предположим, что mobil enet ssd v2 обнаруживает 90 различных классов объектов, я хотел бы добавить еще один класс, чтобы модель обнаруживала 91 другой класс вместо 90 классов. Насколько я понимаю и проверил после выполнения обучения переносу с использованием API обнаружения объектов, я мог обнаруживать только те объекты, которые были в моем наборе данных, и старые классы исчезнут. Так как мне сделать то, что я объяснил?