axinc-ai / ailia-models-unity

Unity version of ailia models repository
45 stars 4 forks source link

ADD detic #68

Closed kyakuno closed 1 year ago

kyakuno commented 1 year ago

ADD detic opset 16

kyakuno commented 1 year ago

https://github.com/axinc-ai/ailia-models/tree/master/object_detection/detic

kyakuno commented 1 year ago

入力はRGB順で0-255。出力は4つ。opset 16の場合、postprocessは不要。

pred_boxes : (27, 4)
scores : (27, )
pred_classes : (27,)
pred_masks : (27, 960, 1280)

num_instancesはlen(boxes)で決まり、boxesはx0,y0,x1,y1で入っている。

kyakuno commented 1 year ago

boxesはピクセル座標系。masksは0-1のレンジのマスク画像がそのまま入っている。

kyakuno commented 1 year ago

cv2.imwrite("mask.jpg", (masks[i]*255).astype(np.uint8)) した画像の例。

mask

kyakuno commented 1 year ago

実装はシンプルにかけそう。

kyakuno commented 1 year ago

別のリポジトリで作業済み。ailia SDK 1.2.14のタイミングでマージ予定。

kyakuno commented 1 year ago

Deticを追加した。