cudasegment.h
Go to the documentation of this file.
00001 /*
00002  * Copyright (c) 2011, Mårten Björkman (celle@csc.kth.se) 
00003  * All rights reserved.
00004  *
00005  * Redistribution and use in source and binary forms, with or without
00006  * modification, are permitted provided that the following conditions are
00007  * met:
00008  *
00009  *  1.Redistributions of source code must retain the above copyright
00010  *    notice, this list of conditions and the following disclaimer.
00011  *  2.Redistributions in binary form must reproduce the above
00012  *    copyright notice, this list of conditions and the following
00013  *    disclaimer in the documentation and/or other materials provided
00014  *    with the distribution.  
00015  *  3.The name of Mårten Björkman may not be used to endorse or
00016  *    promote products derived from this software without specific
00017  *    prior written permission.
00018  *
00019  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00020  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00021  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
00022  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
00023  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
00024  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
00025  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
00026  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
00027  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
00028  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
00029  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00030  */
00031 
00032 #ifndef CUDASEGMENT_H
00033 #define CUDASEGMENT_H
00034 
00035 #include "pyra/tpimage.h"
00039 typedef unsigned char uchar;
00040 class FgBgSegment;
00041 
00043 class CudaSegment 
00044 {
00045 public:
00047   static const int max_figures = 8;
00048 private:
00050   int width;
00052   int height;     
00054   uchar *allocated_d;
00056   uchar *cimd_d;
00058   uchar *vimd_d;
00060   float *dimd_d;
00062   uint *bins_d;          
00064   int *sums_d;                  
00066   float *costh_d;
00068   float *costv_d;        
00070   float *priors_d[max_figures+2]; 
00072   float *beliefs_d[max_figures+2];
00074   float *probs_d[max_figures+2]; 
00075 
00077 
00079   void CreateHistograms(FgBgSegment &segment);
00081 
00085   float ComputeBeta(uchar *image_d);
00087 
00092   void SetGradientCosts(uchar *image_d, float beta, float gamma);
00093 public:  
00095 
00099   CudaSegment(int width, int height);
00100   ~CudaSegment();
00102   void CopyProbsToDevice(Image<float> **probs_h);
00104 
00111   void Execute(FgBgSegment &segment, Image<uchar> &cimg, 
00112                Image<float> &dimg, int numFigures, 
00113                bool download = true);
00114 };
00115 
00116 
00117 #endif // CUDASEGMENT_H
00118  


active_realtime_segmentation
Author(s): Mårten Björkman. Maintained by Jeannette Bohg
autogenerated on Fri Jan 3 2014 12:02:50