search

pytorch / ao

PyTorch native quantization and sparsity for training and inference
BSD 3-Clause "New" or "Revised" License
1.62k stars 181 forks source link

SAM2 AMG cli and other QoL improvements #1336

Closed cpuhrsch closed 6 days ago

cpuhrsch commented 6 days ago

Adds a simple cli to convert image files to masked image files. See python cli.py --help for info on how to use it.

Adds capability that can automatically download a checkpoint when it's missing.

Updates all other configs to work with torchao version of SAM2

pytorch-bot[bot] commented 6 days ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1336

Note: Links to docs will display an error until the docs builds have been completed.

:white_check_mark: No Failures

As of commit fe338deae1872dbf6ed48f967b150610edc56e89 with merge base 7c3c51fd0de33307e43a1769883a348861d6f7c9 (image): :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.