void segment::segment_sequence(const dMatrix src_M, dMatrix& dst_M, Beliefs E, double c, int *num_ccs, int **sizes) { int i, j, t, seqLength, dimH; seqLength = src_M.getWidth(); dimH = E.belStates[0].getLength(); // Build a graph std::vector<edge> edges; for( t=0; t<seqLength-1; t++ ) { edge e(t,t+1,diff_unary(E,t,t+1,dimH)); edges.push_back(e); } // segment universe u(seqLength); // make a disjoint-set forest segment_graph(u, seqLength, seqLength-1, edges, c); *num_ccs = u.num_sets(); int *elt_labels, *seg_sizes, **seg_elts; elt_labels = new int[seqLength]; seg_sizes = new int[u.num_sets()]; seg_elts = new int*[u.num_sets()]; u.get_segments(seqLength, elt_labels, seg_sizes, seg_elts); for(i=0; i<u.num_sets(); i++) printf("[%d]", seg_sizes[i]); printf("\n"); //getchar(); if( sizes ) { delete[] *sizes; *sizes = new int[u.num_sets()]; memcpy(*sizes,seg_sizes,u.num_sets()*sizeof(int)); } // Generate observations for the new layer dst_M.resize(u.num_sets(), src_M.getHeight()); for( i=0; i<src_M.getHeight(); i++ ) for( j=0; j<src_M.getWidth(); j++ ) dst_M.addValue(i,elt_labels[j],src_M.getValue(i,j)); for( i=0; i<dst_M.getHeight(); i++ ) for( j=0; j<dst_M.getWidth(); j++ ) dst_M(i,j) = dst_M(i,j) / seg_sizes[j]; // Clean up and return delete[] elt_labels; elt_labels = 0; delete[] seg_sizes; seg_sizes = 0; for(i=0; i<u.num_sets(); i++) { delete[] seg_elts[i]; seg_elts[i] = 0; } delete[] seg_elts; seg_elts = 0; }
void segment::segment_sequence(const dMatrix src_M, dMatrix& dst_M, std::vector<Beliefs> E, double c, int *num_ccs, int**labels) { int i, j, t, seqLength, dimH, dimY; seqLength = src_M.getWidth(); dimH = E[0].belStates[0].getLength(); dimY = (int)E.size(); std::vector<Beliefs> logE; getLogE(E, logE); // Build a graph std::vector<edge> edges; for( t=0; t<seqLength-1; t++ ) { edge e(t,t+1,diff_unary(E,t,t+1,dimH,dimY)); edges.push_back(e); } // segment universe u(seqLength); // make a disjoint-set forest segment_graph(u, seqLength, seqLength-1, edges, c); *num_ccs = u.num_sets(); int *elt_labels, *seg_sizes, **seg_elts; elt_labels = new int[seqLength]; seg_sizes = new int[u.num_sets()]; seg_elts = new int*[u.num_sets()]; u.get_segments(seqLength, elt_labels, seg_sizes, seg_elts); // Optionally, save elt_labels to labels if( labels!=0 ) memcpy(*labels, elt_labels, sizeof(int)*seqLength); // Generate observations for the new layer dst_M.resize(u.num_sets(), src_M.getHeight()); for( i=0; i<src_M.getHeight(); i++ ) for( j=0; j<src_M.getWidth(); j++ ) dst_M.addValue(i,elt_labels[j],src_M.getValue(i,j)); for( i=0; i<dst_M.getHeight(); i++ ) for( j=0; j<dst_M.getWidth(); j++ ) dst_M(i,j) = dst_M(i,j) / seg_sizes[j]; // Clean up and return delete[] elt_labels; elt_labels = 0; delete[] seg_sizes; seg_sizes = 0; for(i=0; i<u.num_sets(); i++) { delete[] seg_elts[i]; seg_elts[i] = 0; } delete[] seg_elts; seg_elts = 0; }