Views
No views yet

stage1_wholefly_seg.pt: whole-fly segmentation, one class (fly). Finds and segments every fly in the full frame.stage2_bodyparts_seg.pt: body-part segmentation (head, thorax, abdomen), run on single-fly crops taken from stage 1.1from ultralytics import YOLO
2model = YOLO("stage1_wholefly_seg.pt")
3results = model.predict("frame.png")