Metadata-Version: 2.1
Name: pbaa
Version: 1.0.0
Summary: Prompt based automatic annotation
Project-URL: Documentation, https://github.com/dh031200/pbaa#readme
Project-URL: Issues, https://github.com/dh031200/pbaa/issues
Project-URL: Source, https://github.com/dh031200/pbaa
Author-email: dh031200 <imbird0312@gmail.com>
License-Expression: Apache-2.0
License-File: LICENSE.txt
Keywords: Annotation,Grounded-Segment-Anything,Prompt
Classifier: Development Status :: 4 - Beta
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Programming Language :: Python :: Implementation :: PyPy
Requires-Python: >=3.8
Requires-Dist: click
Requires-Dist: loguru
Requires-Dist: wget
Description-Content-Type: text/markdown

# pbaa

[![PyPI - Version](https://img.shields.io/pypi/v/pbaa.svg)](https://pypi.org/project/pbaa)
[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pbaa.svg)](https://pypi.org/project/pbaa)

Easy inference implementation of [Grounded-Sam](https://github.com/IDEA-Research/Grounded-Segment-Anything) for
**Prompt-based automatic annotation**

-----

**Table of Contents**

- [Installation](#installation)
- [Usage](#Usage)
- [License](#license)

## Grounded-SAM

Most of code based on [Grounded-Sam](https://github.com/IDEA-Research/Grounded-Segment-Anything)

## Installation

```console
pip install pbaa
```

## Usage

### Option

* `--source`, `-s` : Path to source image ~~or directory~~. (processing)
    * --source \<Source image>
    * --source source_image.jpg
* `--prompt`, `-p` : Space-separated a pair of prompt and target classe. (Multi)
  * --prompt \"\<Prompt>" \<Class>
  * --prompt "black dog" dog

### CLI

```console
# pbaa -s <Source> -p <prompt> <class> -p <prompt> <class> ...

pbaa -s source_image.jpg -p "black dog" dog
pbaa -s source_image.jpg -p "black dog" dog -p "white cat" cat
```

### Python

```python
from pbaa import inference

inference("path/to/source_image.jpg", {"black dog": "dog", "white cat": "cat"})
```

## License

`pbaa` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.

## Acknowledgements
Grounded-Segment-Anything :[https://github.com/IDEA-Research/Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything)
