DeepGlobe Land Cover Classification Challenge遥感影像语义分割
English | 简体中文
Each satellite image is paired with a mask image for land cover annotation. The mask is a RGB image with 7 classes of labels, using color-coding (R, G, B) as follows.
File names for satellite images and the corresponding mask image are id_sat.jpg and id_mask.png.
Please note:
This repo borrows code heavily from
@INPROCEEDINGS{9064236,
author={Y. {Li} and L. {Chen}},
booktitle={2019 IEEE 5th International Conference on Computer and Communications (ICCC)},
title={Land Cover Classification for High Resolution Remote Sensing Images with Atrous Convolution and BFS}, year={2019},
volume={},
number={},
pages={1808-1813},}
rgb2label.py the satellite images id_mask.png
are RGB images. This code change the RGB images to onechannel images so we can use them to generate tfrecord
files.
create_tf_record_all.py generate tfrecord files. The input and output directory were set in the code follows.
parser.add_argument('--data_dir', type=str, default='./dataset/',
help='Path to the directory containing the PASCAL VOC data.')
parser.add_argument('--output_path', type=str, default='./dataset',
help='Path to the directory to create TFRecords outputs.')
parser.add_argument('--image_data_dir', type=str, default='land_train',
help='The directory containing the image data.')
parser.add_argument('--label_data_dir', type=str, default='onechannel_label',
help='The directory containing the augmented label data.')
train.py
inference.py To apply semantic segmentation to your images.
utils a toolkit
deeplab_model.py the deeplabV3+ model
tensorboard
tensorboard --logdir MODEL_DIR
If you want to run Tensorboard on a remote server. This stackoverflow discussion may be help