• 1000 Я хочу сохранить предварительно обученную модель с моим пользовательским классом как добавленным.
Я устанавливаю num_classes от 90 до 91 в файле конфигурации конвейера. и я объединяю набор данных coco_2017 и свой собственный набор данных и создаю файл tfrecord. Затем я использую model_train.py для переобучения. Окончательная потеря составляет около 10. Я тестирую обученную модель, но она не может обнаружить какой-либо объект где бы то ни было в наборе данных coco или пользовательском наборе данных.
Еще более странно, я переобучаю предварительно обученную модель только с набором данных coco_2017. Результат также не может обнаружить ни одного объекта. Но если использовать предварительно обученную модель, она работает хорошо.
Я также попытался переобучить модель только с моим пользовательским набором данных и установил num_classes на 1. В label_map.pbtxt есть только один элемент. Обученная модель хорошо определяет мой объект.
Для меня это очень странно. Я понятия не имею об этих вопросах. Есть предложения?
Спасибо.