最近一直在做目标检测的项目,C++写程序真的烦,在面对大量图片显示的时候总觉得只用imshow不太好看,所以就做了这么一个可以动态显示图片阵列的程序,可以选择两种输出方式,一种是固定行列数的,一种是设置最大同时显示数量的自动生成行列数的,原理很简单,在这里就不班门弄斧了。
下面是写好的类,如果有需要就直接复制到自己的项目里吧。
class my_show{
public:
my_show(){
current_num=0;
current_show_num=0;
auto_row_num=1;
auto_col_num=1;
is_auto=false;
}
void push(cv::Mat img){mul_img.push_back(img);++current_num;}
void show(int height,int width,int row_num,int col_num){
if(row_num!=-1&&col_num!=-1){mul_show(big_img,mul_img,col_num,row_num,height/row_num,width/col_num);
imshow("my_show",big_img);
waitKey(30);}
else{
std::cout<<"param wrong!"<<std::endl;
exit(0);
}
}
void auto_show(int height,int width,int max_num){
auto_max_num=max_num;
set_auto_num();
mul_show(big_img_auto,mul_img,auto_col_num,auto_row_num,height/auto_row_num,width/auto_col_num);
imshow("auto_show",big_img_auto);
waitKey(30);
}
private:
cv::Mat big_img,big_img_auto;
std::vector<cv::Mat> mul_img;
int current_num;
int current_show_num;
int auto_row_num;
int auto_col_num;
bool is_auto;
int auto_max_num;
void mul_show(cv::Mat &big_img,std::vector<cv::Mat> mul_img,int col_num,int row_num,int row,int col){
int offset_row=10;
int offset_col=10;
big_img.create(row*row_num+offset_row*(row_num+1),col*col_num+offset_col*(col_num+1),mul_img[0].type());
for(int i=0;i<row_num;i++){
for(int k=0;k<col_num;k++){
if(i*row_num+k>=current_num){break;}
cv::Mat tmp;
cv::resize(mul_img[i*row_num+k],tmp,Size(col,row));
tmp.copyTo(big_img(Rect(k*col+offset_col*(k+1),i*row+offset_row*(i+1),col,row)));
if(i*row_num+k==row_num*col_num-1){
if(!is_auto){
clear();
}else{
if(i*row_num+k>=auto_max_num){clear();}
}
}
}
}
}
void clear(){
current_num=0;
mul_img.clear();
}
void set_auto_num(){
is_auto=true;
if(current_num>auto_row_num*auto_col_num){
auto_row_num++;
auto_col_num++;}
}
};