Metadata-Version: 2.1
Name: tjwmd
Version: 1.0.1
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

## Installation

```shell
pip install tjwmd
```

## Quick Start

```python
import cv2
from tjwmd import TJWMD
from ultralytics import YOLO

image = cv2.imread('image.jpg')

wmc = YOLO(
    'wmc_model.pt'
)
wmd = YOLO(
    'wmd_model.pt'
)
tjwmd = TJWMD(
    wmc,
    wmc.names.values(),
    wmd
)

r = tjwmd.predict(
    frame_=image,
    num_of_digits=6,
    angle=0,
    wm_counter_conf=0.4,
    wm_digits_conf=0.1
)

if r:
    results, result_img = r
    print(result_img)
    cv2.imwrite('result.jpg', result_img)

```
