FC2カウンター FPGAの部屋 kv260_median アクセラレーション・プラットフォームに Vitis-AI の DPU を追加する13
fc2ブログ

FPGAやCPLDの話題やFPGA用のツールの話題などです。 マニアックです。 日記も書きます。

FPGAの部屋

FPGAの部屋の有用と思われるコンテンツのまとめサイトを作りました。Xilinx ISEの初心者の方には、FPGAリテラシーおよびチュートリアルのページをお勧めいたします。

kv260_median アクセラレーション・プラットフォームに Vitis-AI の DPU を追加する13

kv260_median アクセラレーション・プラットフォームに Vitis-AI の DPU を追加する12”の続き。

kv260_median アクセラレーション・プラットフォームに Vitis-AI の DPU を追加してみたいということで、前回は、Tensorflowモデルの生成と量子化を行った。今回は、コンパイルと prototxtの用意 を行った。

参考にさせていただくのは、”KV260向けにVitisプラットフォームを作成してDPUを動かす その2 (Vitis 2022.1 + Vitis-AI v2.5)”だ。

kv260_median_platform/kv260_median_pkg/dpuprj_system_hw_link/Hardware/dpu.build/link/vivado/vpl/prj/prj.gen/sources_1/bd/system/ip/system_DPUCZDX8G_1_0/arch.json を
kv260_median_DPU_100_221120.png

DNN/Vitis-AI/kv260_my arch.json としてコピーした。
kv260_median_DPU_101_221120.png

Vitis-AI /workspace/tf_yolov4_coco_416_416_60.3G_2.5 ディレクトリでコンパイルを行う。
vai_c_tensorflow -f ./quantized/quantize_results_416/quantize_eval_model.pb -a /workspace/kv260_myarch.json -o ./compiled -n yolov4_leaky_416_tf
kv260_median_DPU_102_221120.png

コンパイルが成功して、Vitis-AI/tf_yolov4_coco_416_416_60.3G_2.5/compiled ディレクトリに yolov4_leaky_416_tf.xmodel が生成された。
kv260_median_DPU_103_221120.png

yolov4_leaky_416_tf.xmodel を netron で見た。
netron は Web ブラウザからテストすることができる。
kv260_median_DPU_105_221120.png

prototxt を用意する
Vitis-AI/yolov4_leaky_416_tf/yolov4_leaky_416_tf.prototxt を Vitis-AI/tf_yolov4_coco_416_416_60.3G_2.5/compiled ディレクトリにコピーして編集する。
cp ../yolov4_leaky_416_tf/yolov4_leaky_416_tf.prototxt ./compiled/yolov4_leaky_416_tf.prototxt

Vitis-AI/tf_yolov4_coco_416_416_60.3G_2.5/compiled/yolov4_leaky_416_tf.prototxt を編集した。
kv260_median_DPU_104_221120.png

name: "yolov4"
kernel {
    mean: 0.0
    mean: 0.0
    mean: 0.0
    scale: 0.00390625
    scale: 0.00390625
    scale: 0.00390625
}
model_type : YOLOv3
yolo_v3_param {
    num_classes: 80
    anchorCnt: 3
    layer_name: "109"
    layer_name: "101"
    layer_name: "93"
    conf_threshold: 0.3
    nms_threshold: 0.6
    biases: 10 
    biases: 13
    biases: 16
    biases: 30
    biases: 33
    biases: 23
    biases: 30
    biases: 61
    biases: 62
    biases: 45
    biases: 59
    biases: 119
    biases: 116
    biases: 90
    biases: 156
    biases: 198
    biases: 373
    biases: 326
    test_mAP: false
}


現在の Vitis-AI/tf_yolov4_coco_416_416_60.3G_2.5/compiled ディレクトリを示す。
kv260_median_DPU_106_221120.png
  1. 2022年11月21日 05:23 |
  2. Vitis
  3. | トラックバック:0
  4. | コメント:0

コメント

コメントの投稿


管理者にだけ表示を許可する

トラックバック URL
https://marsee101.blog.fc2.com/tb.php/5771-f3e41af3
この記事にトラックバックする(FC2ブログユーザー)