Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
configs		configs
data		data
eval		eval
network		network
utils		utils
.gitignore		.gitignore
README.md		README.md
config.json		config.json
dataset.py		dataset.py
environment.yml		environment.yml
eval_stack.py		eval_stack.py
loss.py		loss.py
module_test.py		module_test.py
train.py		train.py
train_stack.py		train_stack.py

Repository files navigation

Power Grid Mapper

Download the dataset

Electric Transmission and Distribution Infrastructure Imagery Dataset

Environment Setup

conda env create -f environment.yml

Preprocess the data

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

Train the model

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

About

No description, website, or topics provided.

Report repository

Releases

No releases published

Packages

No packages published

Languages