Implementation of U-Net and SegNet for building segmentation

Last update: Dec 07, 2022

Overview

Specialized project

Created by Katrine Nguyen and Martin Wangen-Eriksen as a part of our specialized project at Norwegian University of Science and Technology (NTNU).

Models

Most of our code and the U-net model is significantly inspired by this project Unet-for-Person-Segmentation. The SegNet model we created on our own based on other implementations of SegNet in Tensorflow.

Data

The model is trained and tested on Massachusetts Buildings Dataset from Kaggle. The original images where 1500X1500 pixels each over an area of 1500x1500 meters (1mx1m resolution). The original 137 images were cropped into 64x64 pixels and images without building were filtered out.

To make the masks compatible with our model the masks was changed from white (255,255,255) labels to greyscale with value 1. This is done in image_fix.py found in the repo.

Folder structure

Images and masks are saved in local directories and used in data.py and test.py. This is of course possible to change, however if you want to use the exact same code you can follow this folder structure.


.
├── ...
├── building-segmentation                # Directory for all images
│   ├── Images                           # Directory for raw images
│   │   ├── cropped_images_train_64      # Directory for cropped images where number specifies resolution, containg .jpg
│   │   ├── cropped_images_train_128     # Directory for cropped images where number specifies resolution, containg .jpg 
│   │   └── ...                          # More directories with other resolutions
│   ├── Masks                            # Directory for all maskes
│   │   ├── cropped_masks_train_64       # Directory for cropped masks where number specifies resolution, containg .jpg
│   │   ├── cropped_masks_train_128      # Directory for cropped masks where number specifies resolution, containg .jpg 
│   │   └── ...                          # More directories with other resolutions
│   └── Test                             # Miscellaneous information
│       ├── test_64                      # Directory for images where number specifies resolution, containing .jpg
│       └── ...                          # More directories with other resolutions
└── ...

# data.py
    images = glob(os.path.join(dataset_path, "images/cropped_images_train_64/*"))
    masks = glob(os.path.join(dataset_path, "masks/cropped_masks_train_64/*"))
    
    # In main:
        dataset_path = "building-segmentation"
    
# test.py
    test_images = glob("building-segmentation/test/test_64/*")

Implementation of U-Net and SegNet for building segmentation

Related tags

Overview

Specialized project

Models

Data

Folder structure

Running the project

Requirements

Training

Testing

Owner

Martin.w-e

an Evolutionary Algorithm assisted GAN

A python software that can help blind people find things like laptops, phones, etc the same way a guide dog guides a blind person in finding his way.

Improving Contrastive Learning by Visualizing Feature Transformation, ICCV 2021 Oral

Data and code for the paper "Importance of Kernel Bandwidth in Quantum Machine Learning"

Official code for "On the Frequency Bias of Generative Models", NeurIPS 2021

Official implementation of "Accelerating Reinforcement Learning with Learned Skill Priors", Pertsch et al., CoRL 2020

Code of TIP2021 Paper《SFace: Sigmoid-Constrained Hypersphere Loss for Robust Face Recognition》. We provide both MxNet and Pytorch versions.

FAST Aiming at the problems of cumbersome steps and slow download speed of GNSS data

Evaluation toolkit of the informative tracking benchmark comprising 9 scenarios, 180 diverse videos, and new challenges.

Advanced yabai wooting scripts

Node-level Graph Regression with Deep Gaussian Process Models

Official implementation of the paper Do pedestrians pay attention? Eye contact detection for autonomous driving

MazeRL is an application oriented Deep Reinforcement Learning (RL) framework

rastrainer is a QGIS plugin to training remote sensing semantic segmentation model based on PaddlePaddle.

Implementation of Continuous Sparsification, a method for pruning and ticket search in deep networks

YoHa - A practical hand tracking engine.

A 10000+ hours dataset for Chinese speech recognition

A smart Chat bot that can help to know about corona virus and Make prediction of corona using X-ray.

Using this codebase as a tool for my own research. Making some modifications to the original repo for my own purposes.

SAT: 2D Semantics Assisted Training for 3D Visual Grounding, ICCV 2021 (Oral)