nsubtil / gco-v3.0 Goto Github PK

GCoptimizationGeneralGraph *gc = new GCoptimizationGeneralGraph(nOf,k);
   for(std::vector<double> face : A) 
   {
       for(double neighboor : face)
       {
               int i = iface,j=(int)neighboor;
               Eigen::Vector3f normalp1(N_faces(i,0),N_faces(i,1),N_faces(i,2));
               Eigen::Vector3f normalp2(N_faces(j,0),N_faces(j,1),N_faces(j,2));
               if(j>i)
               {
                   gc->setNeighbors(i,j,edge_cost(normalp1,normalp2,faces_area(i,0),faces_area(j,0)));
               }
       }
       iface++;
   }
   for(int i=0;i<k;i++)
   {
       for(int cur : cluster2Faces[i])
           {
               gc->setLabel(cur,i);
           }
   }
   for(int i=0;i<k;i++)
   {
       for(int j=0;j<nOf;j++)
       {
           gc->setDataCost(j,i,data_term(faces_area(j,0),v1,v2,v3,normalp,normalf,planes[i][3],10));
       }
   }
  for(int i=0;i<k;i++)
   {
       for(int j=0;j<k;j++)
       {
           double weight = 0;
           if(i!=j)
               weight=0.05;
           gc->setSmoothCost(i,j,weight);
       }
   }
   printf("\nBefore optimization energy is %f",gc->compute_energy());
   gc->swap(); 

`