File size: 283 Bytes
4cde79c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
```python
from ultralyticsplus import YOLO, postprocess_classify_output
# load model
model = YOLO('Revrse/icon-labelling')
# set image
image = 'test/155.png'
# perform inference
prediction = model(image)
# observe results
print(prediction[0].names[prediction[0].probs.top1])
``` |