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

PyTorch implementation of probabilistic deep forecast applied to air quality.

Artifacts for paper "MMO: Meta Multi-Objectivization for Software Configuration Tuning"

FasterAI: A library to make smaller and faster models with FastAI.

Simple ray intersection library similar to coldet - succedeed by libacc

LocUNet is a deep learning method to localize a UE based solely on the reported signal strengths from a set of BSs.

给yolov5加个gui界面，使用pyqt5，yolov5是5.0版本

EdiBERT is a generative model based on a bi-directional transformer, suited for image manipulation

Anchor-free Oriented Proposal Generator for Object Detection

Code for the CVPR 2021 paper: Understanding Failures of Deep Networks via Robust Feature Extraction

A concise but complete implementation of CLIP with various experimental improvements from recent papers

PyTorch implementation of Trust Region Policy Optimization

An implementation of the WHATWG URL Standard in JavaScript

Official codes for the paper "Learning Hierarchical Discrete Linguistic Units from Visually-Grounded Speech"

Viewmaker Networks: Learning Views for Unsupervised Representation Learning

This source code is implemented using keras library based on "Automatic ocular artifacts removal in EEG using deep learning"

Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups

Tensorflow implementation of the paper "HumanGPS: Geodesic PreServing Feature for Dense Human Correspondences", CVPR 2021.

ICLR 2021, Fair Mixup: Fairness via Interpolation

一个运行在 𝐞𝐥𝐞𝐜𝐕𝟐𝐏 或 𝐪𝐢𝐧𝐠𝐥𝐨𝐧𝐠 等定时面板的签到项目

Single/multi view image(s) to voxel reconstruction using a recurrent neural network