Segment Anythingは、facebook researchのpytorch実装がありますが、 ScenicにもJax/FlaxのSegment Anythingの実装が公開されています。
READMEにある通りですが、Google Colabでシンプルに実行してみます。
image_pathを変えれば、他の画像でも試せる。
READMEでは、モデル実行時のキーワード引数(input_imageではなくimage)が間違っていて、一部実行エラーになる箇所が あったので、細かい点ではあるがPRを送ってみた。