Torchserve

Room layout recognition

In this project, we focus on building-up an Instance segmentation model for recognizing drawing room layout data. We trained and tested our model (Unet architecture) on both public dataset (Cubicasa, R3D) and own dataset (more than 1000 images of drawing layout) To ensuring generalization of the trained model, we added CVF-FP dataset (public) to test set for evaluation.