mlx-examples/__init__.py at bf2da36fc640e6bfab933ac8c10d76c86fcdb288 - mlx-examples - Gitea: Git with a cup of tea

liuhuayong/mlx-examples

Shiyu 8353bbbf93

Segment Anything Model (#552 )

* add segment anything model

* add readme

* reorg file structure

* update

* lint

* minor updates

* ack

* fix weight loading

* simplify

* fix to run notebooks

* amg in mlx

* remove torch dependency

* nit in README

* return indices in nms

* simplify

* bugfix / simplify

* fix bug'

* simplify

* fix notebook and remove output

* couple more nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-06-02 16:45:51 -07:00

2 lines

64 B

Python

Raw Blame History

from .automatic_mask_generator import SamAutomaticMaskGenerator