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

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


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

Research units

  • CubiCasa Oy


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:


Original languageEnglish
Title of host publicationImage Analysis - 21st Scandinavian Conference, SCIA 2019, Proceedings
EditorsMichael Felsberg, Per-Erik Forssén, Jonas Unger, Ida-Maria Sintorn
Publication statusPublished - 1 Jan 2019
MoE publication typeA4 Article in a 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


ConferenceScandinavian Conference on Image Analysis
Abbreviated titleSCIA

    Research areas

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

ID: 35176950