diff --git a/README.md b/README.md index aacacb1..407b450 100644 --- a/README.md +++ b/README.md @@ -54,7 +54,7 @@ or generate masks for an entire image: ``` from segment_anything import build_sam, SamAutomaticMaskGenerator mask_generator = SamAutomaticMaskGenerator(build_sam(checkpoint="")) -masks = mask_generator_generate() +masks = mask_generator.generate() ``` Additionally, masks can be generated for images from the command line: