Spaces:
Runtime error
Runtime error
| 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) | |