File size: 1,084 Bytes
dbe148f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import json

from ultralytics import YOLO

# Load a model
model = YOLO('best_300.pt')  # load an official model
# model = YOLO('path/to/best_300.pt')  # load a custom model

# Predict with the model
# results = model.predict(source='pCard3', save=True, save_txt=True,project="playing_card",name="predict")
_boxes = []
results = model.predict(source='pCard3/1.jpg', save=True, save_txt=True, project="playing_card", name="predict")
# results = model('https://cdn.britannica.com/23/194523-050-E6C02DBE/selection-American-playing-cards-jack-queen-ace.jpg')
for result in results:
    r = result.numpy()
    names = r.names
    boxes = r.boxes
    for box in boxes:
            b = box.xywh[0].tolist()  # get box coordinates in (top, left, bottom, right) format
            c = int(box.cls[0])
            cf = float(box.conf[0])
            n = names[c]
            _boxes.append({
                "label": c,
                'name': n,
                'probability': cf,
                'bounding': b

            })
j = json.dumps(_boxes)
print(_boxes)