转载——亲测可用!对vector容器中的数据进行滤波!

/***************************************** 
Copyright (c) 2015 Jingshuang Hu 
 
@filename:demo.cpp 
@datetime:2015.12.09 
@author:HJS 
@e-mail:jingshuang_hu@163.com 
@blog:http://blog.csdn.net/hujingshuang 
*****************************************/
 
#include <vector>
#include <algorithm>
#include <iostream>
 
using namespace std;
 
//读数据
bool data_read(vector<int> *vec){
	FILE *fp = fopen("data.txt", "r");
	if (!fp){
		cout<<"open failed!"<<endl;
		return 1;
	}//end if
	int val = 0;
	while(!feof(fp)){
		fscanf(fp, "%d", &val);
		vec->push_back(val);
	}//end while
	fclose(fp);
	return 0;
}
//写数据
bool data_write(vector<int>vec1, vector<int>vec2){
	if (vec1.empty() && vec2.empty()){
		cout<<"mean_filter or median_filter failed!"<<endl;
		return 1;
	}//end if
	FILE *fp;
	fp = fopen("mean_filter.txt", "w");
	vector<int>::iterator it;
	for (it = vec1.begin(); it != vec1.end(); ++it){
		fprintf(fp, "%d\n", *it);
	}//end for
	fclose(fp);
	fp = fopen("median_filter.txt", "w");
	for (it = vec2.begin(); it != vec2.end(); ++it){
		fprintf(fp, "%d\n", *it);
	}//end for
	fclose(fp);
	return 0;
}
//均值滤波
vector<int> mean_filter(vector<int>vec, int scale){
	int sum = 0;
	vector<int>temp;
	for (int i = 0; i < vec.size() - scale; ++i){
		for (int j = 0; j < scale; ++j){
			sum += vec.at(i + j);
		}//end for
		temp.push_back((int)(sum / scale));
		sum = 0;
	}//end for
	return temp;
}
//中值滤波
vector<int> median_filter(vector<int>vec, int scale){
	vector<int>dst, temp;
	dst.resize(scale);
	for (int i = 0; i < vec.size() - scale; ++i){
		partial_sort_copy(vec.begin() + i, vec.begin() + i + scale, dst.begin(), dst.end());
		temp.push_back(dst.at((int)(scale / 2)));
	}//end for
	return temp;
}
//主函数
int main(){
	vector<int>vec, vec1, vec2;
	int scale = 9;
	data_read(&vec);
	vec1 = mean_filter(vec, scale);
	vec2 = median_filter(vec, scale);
	data_write(vec1, vec2);
 
	return 0;
}//end main
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值