axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 325 forks source link

Implement SAM2 video mode #1539

Closed kyakuno closed 2 months ago

kyakuno commented 2 months ago

Change directory

cd ailia-models/image_segmentation/segment-anything-2

ailia SDK 1.4.0

python3 segment-anything-2.py -v demo

ONNX Runtime

python3 segment-anything-2.py --onnx -v demo

1514

kyakuno commented 2 months ago

memory_attentionの4次元バージョンを追加。モデル名をoptにした。

kyakuno commented 2 months ago

torchを全てnumpyに置き換えた。