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
..
utils	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
__init__.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
automatic_mask_generator.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
common.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
image_encoder.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
mask_decoder.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
predictor.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
prompt_encoder.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
sam.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00
transformer.py	Segment Anything Model (#552)	2024-06-02 16:45:51 -07:00