metaseg 0.2.1

No project description provided

MetaSeg: Packaged version of the Segment Anything repository

This repo is a packaged version of the segment-anything model.

Installation

pip install metaseg

Usage

from metaseg import SegAutoMaskGenerator

# If gpu memory is not enough, reduce the points_per_side and points_per_batch.

SegAutoMaskGenerator(
        model_type="vit_h", # "vit_l", "vit_b"
        source= "test.png", # test.mp4
        device="cuda", # "cpu" or "cuda"
        show=True, 
        points_per_side=16, # Optional
        points_per_batch=64, # Optional
)

Extra Features

• Support for video files
• Support for pip installation
• Support for automatic download model weights