Views
No views yet
Queen, queen bee (larger, elongated abdomen)Worker, worker / forager bees (the majority class)Drone, male / drone beesVarroa, varroa destructor mites visible on bees or comb1from ultralytics import YOLO
2from PIL import Image
3
4model = YOLO("honey_bee_detector.pt")
5results = model(Image.open("hive_frame.jpg"), conf=0.10, device="cpu")
6for box in results[0].boxes:
7 cls = model.names[int(box.cls.item())]
8 conf = float(box.conf.item())
9 print(f"{cls}: {conf:.0%}")1PER_CLASS_CONF = {
2 "Worker": 0.25,
3 "Drone": 0.55, # higher, drones false-positive on fingers, shadows
4 "Varroa": 0.30, # mites are small
5 "Queen": 0.15, # lower, try harder to find her
6}yolov8s.pt