Actualice a OpenVINO™ 2020.4.
Consulte La conversión de modelos YOLO* a la representación intermedia (IR) para encontrar los comandos para congelar el modelo y convertirlo a formato IR:
- Congelar el modelo:
python convert_weights_pb.py --class_names coco.names --data_format MEZZANINEWC --weight_file yowc3-tiny.weights –tiny - Convertir el modelo congelado con 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.pbx --batch 1
Dónde:
- --batch define la forma de la entrada del modelo. En el ejemplo, --batch es igual a 1, pero también puede especificar otros enteros mayores que 1.
- --tensorflow_use_custom_operations_config agrega capas de región que faltan al modelo. En la capa IR, la capa Region tiene el nombre RegionYolo.