Saved in:
Bibliographic Details
Main Author: Guo, Qiushi
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.18908
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914850256453632
author Guo, Qiushi
author_facet Guo, Qiushi
contents Detecting obstacles in railway scenarios is both crucial and challenging due to the wide range of obstacle categories and varying ambient conditions such as weather and light. Given the impossibility of encompassing all obstacle categories during the training stage, we address this out-of-distribution (OOD) issue with a semi-supervised segmentation approach guided by optical flow clues. We reformulate the task as a binary segmentation problem instead of the traditional object detection approach. To mitigate data shortages, we generate highly realistic synthetic images using Segment Anything (SAM) and YOLO, eliminating the need for manual annotation to produce abundant pixel-level annotations. Additionally, we leverage optical flow as prior knowledge to train the model effectively. Several experiments are conducted, demonstrating the feasibility and effectiveness of our approach.
format Preprint
id arxiv_https___arxiv_org_abs_2406_18908
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Universal Railway Obstacle Detection System based on Semi-supervised Segmentation And Optical Flow
Guo, Qiushi
Computer Vision and Pattern Recognition
Detecting obstacles in railway scenarios is both crucial and challenging due to the wide range of obstacle categories and varying ambient conditions such as weather and light. Given the impossibility of encompassing all obstacle categories during the training stage, we address this out-of-distribution (OOD) issue with a semi-supervised segmentation approach guided by optical flow clues. We reformulate the task as a binary segmentation problem instead of the traditional object detection approach. To mitigate data shortages, we generate highly realistic synthetic images using Segment Anything (SAM) and YOLO, eliminating the need for manual annotation to produce abundant pixel-level annotations. Additionally, we leverage optical flow as prior knowledge to train the model effectively. Several experiments are conducted, demonstrating the feasibility and effectiveness of our approach.
title A Universal Railway Obstacle Detection System based on Semi-supervised Segmentation And Optical Flow
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.18908