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