Views
No views yet
1from ultralyticsplus import YOLO, postprocess_classify_output
2
3# load model
4model = YOLO('Revrse/icon-labelling')
5
6
7# set image
8image = 'test/155.png'
9
10# perform inference
11prediction = model(image)
12
13# observe results
14print(prediction[0].names[prediction[0].probs.top1])