Power Grid Mapper

September 26, 2021 ยท View on GitHub

Download the dataset

Electric Transmission and Distribution Infrastructure Imagery Dataset

Environment Setup

conda env create -f environment.yml

Preprocess the data

  1. Make TFRecord for FasterRCNN python data/make_patches
  2. Make segmentation map for StackNet python make_angle_map.py

Train the model

  1. FasterRCNN Please refer to TensorFlow Object Detection API
  2. StackNet python train_stack.py

Contents

  1. 1Download the dataset
  2. 2Environment Setup
  3. 3Preprocess the data
  4. 4Train the model