Merge remote-tracking branch 'origin/main'

This commit is contained in:
Elm Forest 2023-04-07 00:52:49 +08:00
commit 7408767146

View File

@ -54,7 +54,7 @@ or generate masks for an entire image:
``` ```
from segment_anything import build_sam, SamAutomaticMaskGenerator from segment_anything import build_sam, SamAutomaticMaskGenerator
mask_generator = SamAutomaticMaskGenerator(build_sam(checkpoint="</path/to/model.pth>")) mask_generator = SamAutomaticMaskGenerator(build_sam(checkpoint="</path/to/model.pth>"))
masks = mask_generator_generate(<your_image>) masks = mask_generator.generate(<your_image>)
``` ```
Additionally, masks can be generated for images from the command line: Additionally, masks can be generated for images from the command line: