Schritte zur Konvertierung des YOLOv4-Keras-Modells in ein IR-Format.
- Befolgen Sie die Anweisungen in Convert YOLOv4 Model in IR und konnten das Modell nicht in das TensorFlow2*-Format konvertieren.
- Bei der Konvertierung von YOLOv4-Darknet zu Keras-Modell wurde TypeError: buffer is too small for requested array Fehler erhalten.
Ändern Sie das Verzeichnis in Modell-Downloader und laden Sie yolo-v4-tf-Modell mit downloader.py Skript herunter:
cd /deployment_tools/tools/model_downloader
sudo python3 downloader.py --name=yolo-v4-tf
Benennen Sie die benutzerdefinierte Gewichtdatei in yomals4.weights um und ersetzen Sie sie in das folgende Verzeichnis:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf
Benennen Sie die benutzerdefinierte Konfigurationsdatei in yo config4.cfg um und ersetzen Sie sie in das folgende Verzeichnis:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf/keras-YOLOv3-model-set/cfg
Wechseln Sie in das nachstehende Verzeichnis und führen Sie pre-convert.py Skript aus.
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 wird im folgenden Verzeichnis generiert :
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf
Ändern Sie das Verzeichnis in Model Optimizer und führen Sie mo.py Skript aus:
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
Die zwischengeschaltete Darstellung wird im folgenden Verzeichnis generiert:
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf/FP16