Segment Anything Model (#552)

Browse Source

* 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>

...

This commit is contained in:

Shiyu

2024-06-03 07:45:51 +08:00

committed by GitHub

parent 89b0b75250

commit 8353bbbf93

22 changed files with 3667 additions and 0 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

BIN

segment_anything/notebooks/images/dog.jpg Normal file

Binary file not shown.

Side by Side After Width:  |  Height:  |  Size: 98 KiB

BIN

segment_anything/notebooks/images/groceries.jpg Normal file

Binary file not shown.

Side by Side After Width:  |  Height:  |  Size: 164 KiB

BIN

segment_anything/notebooks/images/truck.jpg Normal file

Binary file not shown.

Side by Side After Width:  |  Height:  |  Size: 265 KiB