|
|
--- |
|
|
license: agpl-3.0 |
|
|
--- |
|
|
|
|
|
# Open Prices - Price tag detection model |
|
|
|
|
|
This object detection model was trained on images from the Open Prices database to detect price tags of products in shops. |
|
|
|
|
|
It was trained on 206 epochs using Ultralytics library, using yolov11x version, with images resized to 960x960. This model is licensed under the AGPLv3 license. |
|
|
|
|
|
## Weights |
|
|
|
|
|
Weights are available in the weights/ directory. An ONNX export of the model is available in weights/model.onnx. |
|
|
|
|
|
## Run the model |
|
|
|
|
|
``` |
|
|
# Load a model |
|
|
model = YOLO(f"{data_path}/yolo11x.pt") |
|
|
|
|
|
image_test_list = [f"{data_path}/test/4xXuBb8u3E.jpg", f"{data_path}/test/test_image.jpg"] |
|
|
|
|
|
results = model.predict(image_test_list, conf=0.1, save=True, project=f"{data_path}/test") # or with a list of image paths |
|
|
``` |
