您的位置:首页 > 编程语言 > Java开发

Java 如何实现bwlabel

2016-02-29 12:26 495 查看
       public static Map<String,Object> bwlabel(Mat mat,int n) 
{
Map<String,Object> hash = new HashMap<String,Object>();
int rows = (int) mat.size().height;
int cols = (int) mat.size().width;
Mat labelMap = Mat.zeros(mat.size(),CvType.CV_32SC1);
int labelIndex = 0;
Mat queue = Mat.zeros(rows*cols,2, CvType.CV_32SC1);
int be = 0;
int en = 0;
for(int i = 0; i < rows; i++)
{
for(int j = 0; j < cols; j++)
{
if(MatUtils.getIntData(labelMap,i,j) != 0 || MatUtils.getIntData(mat,i,j) == 0)
{
continue;
}
labelIndex =labelIndex + 1;
labelMap.put(i,j,labelIndex);
//上
if( i > 1 && MatUtils.getIntData(labelMap,i-1,j)== 0 && MatUtils.getIntData(mat,i-1,j) == 1)
{
   labelMap.put(i-1,j,labelIndex); 
   queue.put(en, 0, i-1);
   queue.put(en, 1, j);
   en = en + 1;
}
//下
if(i < rows && MatUtils.getIntData(labelMap, i+1, j) == 0 && MatUtils.getIntData(mat, i+1, j) == 1)
{
           labelMap.put(i+1,j,labelIndex);
           queue.put(en, 0, i+1);
           queue.put(en, 1, j);
           en = en + 1;
}
//左
if(j > 1&& MatUtils.getIntData(labelMap,i,j-1)== 0 && MatUtils.getIntData(mat,i,j-1) == 1)
{
       labelMap.put(i,j-1,labelIndex);
       queue.put(en,0,i);
       queue.put(en,1,j-1);
       en = en + 1;
   }
//右 
if(j < cols && MatUtils.getIntData(labelMap,i,j+1)== 0 && MatUtils.getIntData(mat,i,j+1) == 1)
{

                    labelMap.put(i,j+1, labelIndex);

                    queue.put(en,0,i);
           queue.put(en,1,j+1);
           en = en + 1;
}
if(n == 8)
{
//左上
if(i > 1 && j > 1 && MatUtils.getIntData(labelMap, i-1,j-1) == 0 && MatUtils.getIntData(mat,i-1,j-1) == 1)
{

                        labelMap.put(i-1,j-1, labelIndex);

                        queue.put(en,0,i-1);

                        queue.put(en,1,j-1);
           en = en + 1;
}
//右上
if(i > 1 && j < cols && MatUtils.getIntData(labelMap,i-1,j+1) == 0 && MatUtils.getIntData(mat,i-1,j+1) == 1)
{
       labelMap.put(i-1,j+1,labelIndex);
       queue.put(en,0,i-1);
       queue.put(en,1,j+1);
       en = en + 1;
}
//左下
if(i < rows && j > 1 && MatUtils.getIntData(labelMap,i+1,j-1) == 0 && MatUtils.getIntData(mat,i+1,j-1) == 1)
{
           labelMap.put(i+1,j-1,labelIndex);
           queue.put(en,0,i+1);
       queue.put(en,1,j-1);
           en = en + 1;
}
//右下
if(i < rows && j < cols && MatUtils.getIntData(labelMap,i+1,j+1) == 0 && MatUtils.getIntData(mat,i+1,j+1) == 1)
{
        labelMap.put(i+1,j+1,labelIndex);
        queue.put(en,0,i+1);
    queue.put(en,1,j+1);
        en = en + 1;
}
}
int qi = 0;
int qj = 0;
while(be < en)
{
           qi = MatUtils.getIntData(queue, be, 0);
           qj = MatUtils.getIntData(queue, be, 1);
           be = be + 1;
         //上
               if(qi > 1 && MatUtils.getIntData(labelMap,qi-1,qj)== 0 && MatUtils.getIntData(mat,qi-1,qj) == 1)
               {
                   labelMap.put(qi-1,qj,labelIndex);
                   queue.put(en,0, qi-1);
                   queue.put(en,1, qj);
                   en = en + 1;
               }
               //下
               if(qi < rows && MatUtils.getIntData(labelMap,qi+1,qj)== 0 && MatUtils.getIntData(mat,qi+1,qj) == 1)
               {
                  
                   labelMap.put(qi+1,qj,labelIndex);
                   queue.put(en,0,qi+1);
                   queue.put(en,1,qj);
                   en = en + 1;
               }
               //左
               if(qj >1 &&MatUtils.getIntData(labelMap,qi,qj-1)== 0 && MatUtils.getIntData(mat,qi,qj-1) == 1)
               {
                   labelMap.put(qi,qj-1,labelIndex);
                   queue.put(en,0,qi);
                   queue.put(en,1,qj-1);
                   en = en + 1;
               }
               //右
               if(qj < cols && MatUtils.getIntData(labelMap,qi,qj+1)== 0 && MatUtils.getIntData(mat,qi,qj+1) == 1)
               {
                   labelMap.put(qi,qj+1,labelIndex);
                   queue.put(en,0,qi);
                   queue.put(en,1,qj+1);
                   en = en + 1;
               }
               if(n == 8)
               {
               
//左上
               
if(qi >1 && qj > 1 && MatUtils.getIntData(labelMap,qi-1,qj-1) == 0 && MatUtils.getIntData(mat,qi-1,qj-1) == 1)
               
{
                        labelMap.put(qi-1,qj-1,labelIndex);
                        queue.put(en,0,qi-1);
                        queue.put(en,1,qj-1);
                        en = en + 1;
               
}

                         //右上
                    if(qi > 1 && qj < cols && MatUtils.getIntData(labelMap,qi-1,qj+1) == 0 && MatUtils.getIntData(mat,qi-1,qj+1) == 1)
                    {
                        labelMap.put(qi-1,qj+1,labelIndex);
                        queue.put(en,0,qi-1);
                        queue.put(en,1,qj+1);
                        en = en + 1;
                    }
                    //左下
                    if(qi < rows && qj > 1 && MatUtils.getIntData(labelMap,qi+1,qj-1) == 0 && MatUtils.getIntData(mat,qi+1,qj-1) == 1)
                    {                         
                   
labelMap.put(qi+1,qj-1,labelIndex);
                   
queue.put(en,0,qi+1);
                        queue.put(en,1,qj-1);
                        en = en + 1;
                    }
                    //右下 
                    if(qi < rows && qj < cols && MatUtils.getIntData(labelMap,qi+1,qj+1) == 0 && MatUtils.getIntData(mat,qi+1,qj+1) == 1)
                    {
                        labelMap.put(qi+1,qj+1,labelIndex);
                   
queue.put(en,0,qi+1);
                        queue.put(en,1,qj+1);
                        en = en + 1;
                    }    
               }
}


}
hash.put("Mat",labelMap);
hash.put("Num", labelIndex);
return hash;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息