SSTD: Stripe-Like Space Target Detection Using Single-Point Weak Supervision

TL;DR

This paper introduces a new dataset and a teacher-student framework for detecting stripe-like space targets with minimal supervision, achieving state-of-the-art results and strong zero-shot generalization.

Contribution

The paper presents AstroStripeSet, a pioneering dataset, and a novel label evolution framework with single-point weak supervision for space target detection.

Findings

Achieves performance comparable to fully supervised methods

Exhibits strong zero-shot generalization on real-world images

Sets new state-of-the-art benchmark in SSTD

Abstract

Stripe-like space target detection (SSTD) plays a key role in enhancing space situational awareness and assessing spacecraft behaviour. This domain faces three challenges: the lack of publicly available datasets, interference from stray light and stars, and the variability of stripe-like targets, which makes manual labeling both inaccurate and labor-intensive. In response, we introduces `AstroStripeSet', a pioneering dataset designed for SSTD, aiming to bridge the gap in academic resources and advance research in SSTD. Furthermore, we propose a novel teacher-student label evolution framework with single-point weak supervision, providing a new solution to the challenges of manual labeling. This framework starts with generating initial pseudo-labels using the zero-shot capabilities of the Segment Anything Model (SAM) in a single-point setting. After that, the fine-tuned StripeSAM serves as the teacher and the newly developed StripeNet as the student, consistently improving segmentation performance through label evolution, which iteratively refines these labels. We also introduce `GeoDice', a new loss function customized for the linear characteristics of stripe-like targets. Extensive experiments show that our method matches fully supervised approaches, exhibits strong zero-shot generalization for diverse space-based and ground-based real-world images, and sets a new state-of-the-art (SOTA) benchmark. Our AstroStripeSet dataset and code will be made publicly available.