ID do artigo: 000088649 Tipo de conteúdo: Solução de problemas Última revisão: 20/05/2022

Não é possível converter o modelo YOLOv4 Keras* para um formato de representação intermediária (IR)

BUILT IN - ARTICLE INTRO SECOND COMPONENT
Resumo

Etapas para converter o modelo YoLOv4 Keras em um formato IR.

Descrição
  • Seguiu a instrução em Converter o modelo YOLOv4 para IR e não converteu o modelo em formato TensorFlow2*.
  • Erro TypeError: buffer is too small for requested array ao converter a darknet YOLOv4 para o modelo Keras.
Resolução

Mude o diretório para Model Downloader e baixe o modelo yolo-v4-tf usando downloader.py script:
cd /deployment_tools/tools/model_downloader
sudo python3 downloader.py --name=yolo-v4-tf


Renomeie o arquivo de peso personalizado para yolov4.weights e substitua o seguinte diretório:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf

Renomeie o arquivo de configuração personalizado para yolov4.cfg e substitua o seguinte diretório:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf/keras-YOLOv3-model-set/cfg

Mude para o diretório a seguir e execute pre-convert.py script.
cd /opt/intel/openvino_2021/deployment_tools/open_model_zoo/models/public/yolo-v4-tf
sudo python3 pre-convert.py "/deployment_tools/tools/model_downloader/public/yolo-v4-tf" "/deployment_tools/tools/model_downloader/public/yolo-v4-tf"


yolo-v4.savedmodel será gerado no seguinte diretório:
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf

Mude o diretório para otimizador de modelo e execute mo.py script:
cd /opt/intel/openvino_2021/deployment_tools/model_optimizer
sudo python3 mo.py --framework=tf --data_type=FP16 --output_dir=/deployment_tools/tools/model_downloader/public/yolo-v4-tf/FP16 --model_name=yolo-v4-tf --input_shape=[1,416,416,3] --input=image_input --scale_values=image_input[255] --reverse_input_channels --saved_model_dir=/deployment_tools/tools/model_downloader/public/yolo-v4-tf/yolo-v4.savedmodel


A Representação Intermediária será gerada no seguinte diretório:
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf/FP16

Produtos relacionados

Este artigo aplica-se a 2 produtos

O conteúdo desta página é uma combinação de tradução humana e por computador do conteúdo original em inglês. Este conteúdo é fornecido para sua conveniência e apenas para informação geral, e não deve ser considerado completo ou exato. Se houver alguma contradição entre a versão em inglês desta página e a tradução, a versão em inglês prevalecerá e será a determinante. Exibir a versão em inglês desta página.