from ultralytics import YOLO
import cv2
import matplotlib.pyplot as plt

# Load the YOLO model
model = YOLO('yolo11x_leaf.pt')

# Run inference on an image or directory
result = model.predict('file/directory', task="detect", save=False, conf=0.15)

# Load the original image
image_path = result.path
image = cv2.imread(image_path)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# Annotate the image with predictions
annotated_image = result.plot()

# Display the annotated image
plt.figure(figsize=(10, 7))
plt.imshow(annotated_image)
plt.axis("off")
plt.title(f"Predictions for Image")
plt.show()

pedromiguelsanchez/yolo-plant-leaf-detection

YOLOv11 Model for Plant Leaves Detection

Model Details

Dataset

Usage

Examples

pedromiguelsanchez/
yolo-plant-leaf-detection