OneGrow: Unified Temporal Plant Image and Mask Generation

Image-based crop phenotyping benefits from image series that capture plant development together with organ-level labels. Such paired data are limited because organ annotation is expensive, and following the same plants over time requires repeated, registered imaging. Existing generative models for plants either synthesize temporal imagery without structural labels or generate labeled images without a temporal dimension. We introduce OneGrow, a latent flow-matching model that jointly models wheat images and their organ-segmentation masks over time. Images and masks share a single frozen image autoencoder. A reveal specifies which content is observed context, so the same model covers tasks such as mask-to-image synthesis, image-to-mask segmentation, and temporal forecasting. For sequences longer than the training window, a sliding-window roll-out generates each new image from the preceding frames, keeping long sequences temporally consistent. We train jointly on a large single-frame wheat dataset and a multi-year temporal dataset, using pseudo-labels from a pretrained segmentation model. We evaluate segmentation and image quality against held-out references and assess the multi-task and temporal behavior qualitatively.