Java 如何实现bwlabel

       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;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值