遥感语义分割代码

基于深度学习对影像进行语义分割的算法具体实现，包含常见的Unet,SEGNET等深度学习语义分割模型，实现框架为Keras

#include “opencv2/imgproc.hpp“
#include “opencv2/highgui.hpp“


using namespace cv;

#define ALL 0
#define VEGETATION 1
#define ROAD 4
#define BUILDING 2
#define WATER 3

int main（int argc char* argv[]）
{
	if （argc != 3）
	{
		printf（“invalid parameters! Usage: [input_src] [input_mask]\n“）;
		return -1;
	}

	string input_src = argv[1];
	string input_mask = argv[2];


	Mat mask = imread（input_mask 0）;
	Mat src = imread（input_src）;
	for （int i = 0; i < src.rows; i++）
	{
		for （int j = 0; j < src.cols; j++）
		{
			uchar *p_mask = mask.ptr（i j）;
			uchar *p_src = src.ptr（i j）;

			if （p_mask[0] == VEGETATION）
			{
				p_src[0] = 159;
				p_src[1] = 255;
				p_src[2] = 84;
			}
			else if （p_mask[0] == ROAD）
			{
				p_src[0] = 38;
				p_src[1] = 71;
				p_src[2] = 139;
			}
			else if （p_mask[0] == BUILDING || p_mask[0] == 255）
			{
				p_src[0] = 34;
				p_src[1] = 180;
				p_src[2] = 238;
			}
			else if （p_mask[0] == WATER）
			{
				p_src[0] = 255;
				p_src[1] = 191;
				p_src[2] = 0;
			}


		}
	}

	imwrite（“stack.png“ src）;


	return 0;
}

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-04-03 04:24  Satellite-Segmentation-master\
     文件         378  2018-04-03 04:24  Satellite-Segmentation-master\.gitattributes
     文件         574  2018-04-03 04:24  Satellite-Segmentation-master\.gitignore
     文件       26710  2018-04-03 04:24  Satellite-Segmentation-master\1.png
     文件      453242  2018-04-03 04:24  Satellite-Segmentation-master\2.png
     文件       73014  2018-04-03 04:24  Satellite-Segmentation-master\3.png
     文件       42182  2018-04-03 04:24  Satellite-Segmentation-master\4.png
     文件        3096  2018-04-03 04:24  Satellite-Segmentation-master\README.md
     目录           0  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\
     文件       15837  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\all_net.py
     文件         601  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\correct_labels.cpp
     文件        2682  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\predict.py
     文件       20132  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\segnet.py
     文件       11836  2018-04-03 04:24  Satellite-Segmentation-master\deprecated\unet.py
     文件        1074  2018-04-03 04:24  Satellite-Segmentation-master\draw_lables.cpp
     目录           0  2018-04-03 04:24  Satellite-Segmentation-master\ensemble\
     文件         921  2018-04-03 04:24  Satellite-Segmentation-master\ensemble\vote.py
     目录           0  2018-04-03 04:24  Satellite-Segmentation-master\segnet\
     文件        3111  2018-04-03 04:24  Satellite-Segmentation-master\segnet\gen_dataset.py
     文件        2545  2018-04-03 04:24  Satellite-Segmentation-master\segnet\segnet_predict.py
     文件       10273  2018-04-03 04:24  Satellite-Segmentation-master\segnet\segnet_train.py
     目录           0  2018-04-03 04:24  Satellite-Segmentation-master\unet\
     文件        2439  2018-04-03 04:24  Satellite-Segmentation-master\unet\combind.py
     文件        3128  2018-04-03 04:24  Satellite-Segmentation-master\unet\gen_dataset.py
     文件        2432  2018-04-03 04:24  Satellite-Segmentation-master\unet\unet_predict.py
     文件        7469  2018-04-03 04:24  Satellite-Segmentation-master\unet\unet_train.py