Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

  1. data/cnn/*.bin をご使用のモデルに置き換えてください。

  2. main.cppの以下の部分をモデルの仕様に合わせて変更してください。

    #define OUTSIZE_HEIGHT <モデルの入力高さ>
    #define OUTSIZE_WIDTH <モデルの入力幅>
    #define NETNAME <モデルのファイル名>
    #define LAYERNAMEIN <モデルの入力レイヤー>
    #define LAYERNAMEOUT_CONV1037 <特徴マップ(ストライド 32)のレイヤー名>
    #define LAYERNAMEOUT_CONV1017 <特徴マップ(ストライド 16)のレイヤー名>
    #define LAYERNAMEOUT_CONV997 <特徴マップ(ストライド 8)のレイヤー名>
    #define TOTAL_CLS <モデルのクラス数>

    char const *names (pymain.py では object_name_list)
    モデルの仕様に合わせて変更してください
    const float anchors (pymain.py では anchors_list)
    以下URLの、 anchors の値を使用しています。
    https://github.com/ultralytics/yolov5/blob/master/models/yolov5s.yaml
    モデルによっては値が異なることがありますので、モデルの仕様をご確認ください。

...