При проверке модели Keras yolov3-tiny.h5 с использованием netron
я вижу, что входной узел называется input_1 и имеет тип float32 [?,?,?, 3].Я ожидаю, что float32 [?, 416,416,3]
Как я могу заставить его быть float32 [?, 416,416,3]?Это необходимо для последующей обработки.Модель Keras должна быть преобразована в frozen_model.pb в Tensorflow и затем обработана для развертывания.Инструменты развертывания не могут обрабатывать ввод с неизвестным размером w, h.
Вот как я сгенерировал модель Keras.Я загрузил yolov3-tiny.cfg (https://github.com/pjreddie/darknet/blob/master/cfg/yolov3-tiny.cfg) и yolov3-tiny.weigths (https://pjreddie.com/media/files/yolov3-tiny.weights)
) и затем преобразовал модель в модель keras с помощью следующей команды:
python convert.py -p yolov3-tiny.cfg yolov3-tiny.weights model_data / yolov3-tiny.h5
(этот код получается путем клонирования https://github.com/qqwweee/keras-yolo3)
Создание прогноза с использованиемсохраненная модель Keras работает нормально: python yolo_video.py --image --model model_data / yolov3-tiny.h5
Однако при проверке модели Keras yolov3-tiny.h5 с помощью netron
Iвижу, что входной узел называется input_1 и имеет тип float32 [?,?,?, 3]
Я ожидаю, что float32 [?, 416,416,3]
Как я могу заставить егоbe float32 [?, 416,416,3]?