Supported Labels
['Black-grass', 'Charlock', 'Cleavers', 'Common Chickweed', 'Common wheat', 'Fat Hen', 'Loose Silky-bent', 'Maize', 'Scentless Mayweed', 'Shepherds Purse', 'Small-flowered Cranesbill', 'Sugar beet']
How to use
pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
- Load model and perform prediction:
from ultralyticsplus import YOLO, postprocess_classify_output
model = YOLO('uisikdag/weed_yolov8_balanced')
model.overrides['conf'] = 0.25
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
results = model.predict(image)
print(results[0].probs)
processed_result = postprocess_classify_output(model, result=results[0])
print(processed_result)