ContRail: A Framework for Realistic Railway Image Synthesis using ControlNet

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Alexandrescu, Andrei-Robert, Petec, Razvan-Gabriel, Manole, Alexandru, Diosan, Laura-Silvia
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915057020960768
author Alexandrescu, Andrei-Robert
Petec, Razvan-Gabriel
Manole, Alexandru
Diosan, Laura-Silvia
author_facet Alexandrescu, Andrei-Robert
Petec, Razvan-Gabriel
Manole, Alexandru
Diosan, Laura-Silvia
contents Deep Learning became an ubiquitous paradigm due to its extraordinary effectiveness and applicability in numerous domains. However, the approach suffers from the high demand of data required to achieve the potential of this type of model. An ever-increasing sub-field of Artificial Intelligence, Image Synthesis, aims to address this limitation through the design of intelligent models capable of creating original and realistic images, endeavour which could drastically reduce the need for real data. The Stable Diffusion generation paradigm recently propelled state-of-the-art approaches to exceed all previous benchmarks. In this work, we propose the ContRail framework based on the novel Stable Diffusion model ControlNet, which we empower through a multi-modal conditioning method. We experiment with the task of synthetic railway image generation, where we improve the performance in rail-specific tasks, such as rail semantic segmentation by enriching the dataset with realistic synthetic images.
format Preprint
id arxiv_https___arxiv_org_abs_2412_06742
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ContRail: A Framework for Realistic Railway Image Synthesis using ControlNet
Alexandrescu, Andrei-Robert
Petec, Razvan-Gabriel
Manole, Alexandru
Diosan, Laura-Silvia
Computer Vision and Pattern Recognition
Artificial Intelligence
Deep Learning became an ubiquitous paradigm due to its extraordinary effectiveness and applicability in numerous domains. However, the approach suffers from the high demand of data required to achieve the potential of this type of model. An ever-increasing sub-field of Artificial Intelligence, Image Synthesis, aims to address this limitation through the design of intelligent models capable of creating original and realistic images, endeavour which could drastically reduce the need for real data. The Stable Diffusion generation paradigm recently propelled state-of-the-art approaches to exceed all previous benchmarks. In this work, we propose the ContRail framework based on the novel Stable Diffusion model ControlNet, which we empower through a multi-modal conditioning method. We experiment with the task of synthetic railway image generation, where we improve the performance in rail-specific tasks, such as rail semantic segmentation by enriching the dataset with realistic synthetic images.
title ContRail: A Framework for Realistic Railway Image Synthesis using ControlNet
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2412.06742