ayzeksalimli/tram-dynamic-roi-tracker-yolo11s

tram-dynamic-roi-tracker

Graduation Thesis (VKR): Algorithm for detecting foreign objects on tram tracks based on their segmentation.

Seven segmentation models were researched and trained (YOLO-seg: 8n, 8s, 11s, 26s; SegFormer: B0, B3; DeepLabV3+: ResNet50), as well as three detectors: YOLOX, YOLOv8, and YOLOv5.

Based on the research results, the following were selected for the task:

• —Segmentor: YOLOv11s-seg
• —Detector: YOLOv8s
• —Tracker: ByteTrack

The developed algorithm is available at: https://github.com/MathematicLove/tram-dynamic-roi-tracker

Useful Links

• —Research paper (article): https://elibrary.ru/qfcwed
• —Diploma for 1st degree laureate (Best report in ML and Intelligent Data Processing section): Certificate
• —Full Thesis (VKR): coming soon

Available Models