Ошибка dims на входе snpe-tenorflow-to-dlc
Команда ниже используется для преобразования .pb
модель обучена TensorFlow для .dlc
, В соответствии с документацией snpe размер пакета не требуется, однако произошла неясная ошибка.
snpe-tensorflow-to-dlc --graph frozen_defect.pb --input_dim Placeholder 224,224,3 --dlc out.dlc --out_node prediction
Кроме того, я не думаю, что размер пакета должен быть включен в файл модели, но ошибка ниже показывает, что это необходимо.
ERROR - Conversion failed: ERROR_TF_INVALID_INPUT_DIMS: Invalid input dimensions: Shapes (4, 224, 224, 3) and (1, 224, 224, 3) are not compatible
Что я должен делать?