Metadata-Version: 2.1
Name: tjwmd
Version: 2.0.0
Summary: tjwmd
Home-page: https://github.com/duynguyen02/tjwmd
Author: Duy Nguyen
Author-email: duynguyen02.dev@gmail.com
Requires-Python: >=3.9,<4.0
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Dist: opencv-python (>=4.10.0.84,<5.0.0.0)
Requires-Dist: ultralytics (>=8.2.83,<9.0.0)
Project-URL: Repository, https://github.com/duynguyen02/tjwmd
Description-Content-Type: text/markdown

# tjwmd

![PyPI - Version](https://img.shields.io/pypi/v/tjwmd)

## Installation

```shell
pip install tjwmd
```

## Quick Start

```python
from PIL import Image

from tjwmd import TJWMD
from tjwmd.digits_labels import DigitsLabels

dl = DigitsLabels()
dl[0] = ['0', 'Zero']

wmd = TJWMD(
    wm_counter_model_path='yolov8_wmc_v1.pt',
    wm_digits_model_path='yolov8_wmd_v3s.pt',
    digits_labels=dl,
    counter_labels=['counter']
)

img = Image.open('6724502e2cec9d702006b6cc.jpeg')

values, bbox_img = wmd.predict(
    _image=img,
    num_of_digits=6,
    wm_digits_conf=0.1,
    wm_counter_conf=0.1,
    angle=0.0
)
print(values)

bbox_img.show()
```
