Closed kyakuno closed 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で入っている。
boxesはピクセル座標系。masksは0-1のレンジのマスク画像がそのまま入っている。
cv2.imwrite("mask.jpg", (masks[i]*255).astype(np.uint8)) した画像の例。
実装はシンプルにかけそう。
別のリポジトリで作業済み。ailia SDK 1.2.14のタイミングでマージ予定。
Deticを追加した。
ADD detic opset 16