icon-labelling / README.md
Revrse's picture
Rename readme.md to README.md
5ad5407 verified
|
raw
history blame
283 Bytes
```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])
```