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

Cryptocurrency Prediction with Artificial Intelligence (Deep Learning via LSTM Neural Networks)

Pre-trained NFNets with 99% of the accuracy of the official paper

VACA: Designing Variational Graph Autoencoders for Interventional and Counterfactual Queries

Pure python implementations of popular ML algorithms.

Python wrapper class for OpenVINO Model Server. User can submit inference request to OVMS with just a few lines of code

Keras implementation of Deeplab v3+ with pretrained weights

An implementation of the methods presented in Causal-BALD: Deep Bayesian Active Learning of Outcomes to Infer Treatment-Effects from Observational Data.

The first dataset of composite images with rationality score indicating whether the object placement in a composite image is reasonable.

KoRean based ELECTRA pre-trained models (KR-ELECTRA) for Tensorflow and PyTorch

ILVR: Conditioning Method for Denoising Diffusion Probabilistic Models (ICCV 2021 Oral)

git《Tangent Space Backpropogation for 3D Transformation Groups》(CVPR 2021) GitHub:1]

Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.

Sequence-tagging using deep learning

This is the official code for the paper "Tracker Meets Night: A Transformer Enhancer for UAV Tracking".

FwordCTF 2021 Infrastructure and Source code of Web/Bash challenges

Official Code for "Constrained Mean Shift Using Distant Yet Related Neighbors for Representation Learning"

Hybrid CenterNet - Hybrid-supervised object detection / Weakly semi-supervised object detection

Python package to generate image embeddings with CLIP without PyTorch/TensorFlow

Code for EMNLP2020 long paper: BERT-Attack: Adversarial Attack Against BERT Using BERT

Source code of our work: "Benchmarking Deep Models for Salient Object Detection"