Atualize para OpenVINO™ 2020.4.
Consulte Converter modelos YOLO* para a Representação Intermediária (IR) para encontrar os comandos para congelar o modelo e convertê-lo em formato IR:
- Congele o modelo:
python convert_weights_pb.py --class_names coco.names --data_format NHWC -weight_file yolov3-tiny.weights - minúsculo - Converter o modelo congelado para o formato IR:
python3 mo_tf.py --input_model /path/to/yolo_v3_tiny.pb --tensorflow_use_custom_operations_config $MO_ROOT/extensions/front/tf/yolo_v3_tiny.json --lote 1
onde:
- --lote define a forma da entrada do modelo. No exemplo, --batch é igual a 1, mas você também pode especificar outros inteiros maiores que 1.
- --tensorflow_use_custom_operations_config adiciona camadas de região ausentes ao modelo. No IR, a camada Região tem o nome RegionYolo.