CubiCasa5K: A Dataset and an Improved Multi-task Model for Floorplan Image Analysis

Ahti Kalervo*, Juha Ylioinas, Markus Häikiö, Antti Karhu, Juho Kannala

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference article in proceedingsScientificpeer-review

51 Citations (Scopus)

Abstract

Better understanding and modelling of building interiors and the emergence of more impressive AR/VR technology has brought up the need for automatic parsing of floorplan images. However, there is a clear lack of representative datasets to investigate the problem further. To address this shortcoming, this paper presents a novel image dataset called CubiCasa5K, a large-scale floorplan image dataset containing 5000 samples annotated into over 80 floorplan object categories. The dataset annotations are performed in a dense and versatile manner by using polygons for separating the different objects. Diverging from the classical approaches based on strong heuristics and low-level pixel operations, we present a method relying on an improved multi-task convolutional neural network. By releasing the novel dataset and our implementations, this study significantly boosts the research on automatic floorplan image analysis as it provides a richer set of tools for investigating the problem in a more comprehensive manner. Data and code at: https://github.com/CubiCasa/CubiCasa5k.

Original languageEnglish
Title of host publicationImage Analysis - 21st Scandinavian Conference, SCIA 2019, Proceedings
EditorsMichael Felsberg, Per-Erik Forssén, Jonas Unger, Ida-Maria Sintorn
PublisherSpringer
Pages28-40
Number of pages13
ISBN (Print)9783030202040
DOIs
Publication statusPublished - 1 Jan 2019
MoE publication typeA4 Conference publication
EventScandinavian Conference on Image Analysis - Norrköping, Sweden
Duration: 11 Jun 201913 Jun 2019
Conference number: 21

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11482 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceScandinavian Conference on Image Analysis
Abbreviated titleSCIA
Country/TerritorySweden
CityNorrköping
Period11/06/201913/06/2019

Keywords

  • Convolutional neural networks
  • Dataset
  • Floorplan images
  • Multi-task learning

Fingerprint

Dive into the research topics of 'CubiCasa5K: A Dataset and an Improved Multi-task Model for Floorplan Image Analysis'. Together they form a unique fingerprint.

Cite this