Langkah-langkah untuk mengonversi model YOLOv4 Keras ke format IR.
- Instruksi diikuti dalam Mengonversi MODEL YOLOv4 ke IR dan gagal mengonversi model menjadi format TensorFlow2*.
- Menerima TypeError: buffer is too small for requested array galat saat mengonversi darknet YOLOv4 ke model Keras.
Ubah direktori ke Pengunduh Model dan unduh model yolo-v4-tf menggunakan skrip downloader.py:
cd /deployment_tools/tools/model_downloader
sudo python3 downloader.py --name=yolo-v4-tf
Ganti nama file berat kustom menjadi yolov4.weights dan ganti ke direktori berikut:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf
Ganti nama berkas konfigurasi kustom ke yolov4.cfg dan ganti ke direktori berikut:
/deployment_tools/tools/model_downloader/public/yolo-v4-tf/keras-YOLOv3-model-set/cfg
Ubah ke direktori berikut dan jalankan skrip pre-convert.py.
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 akan dihasilkan dalam direktori berikut:
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf
Ubah direktori ke Pengoptimal Model dan jalankan skrip mo.py:
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
Representasi Menengah akan dihasilkan dalam direktori berikut:
/opt/intel/openvino_2021/deployment_tools/tools/model_downloader/public/yolo-v4-tf/FP16