#include <object_segmenter.h>
Classes | |
struct | Action |
struct | Box2D |
Public Types | |
enum | ActionType { PAUSE, RESET, CLICK, REGION, STOP, GRADCHANGE } |
Public Member Functions | |
bool | getCurrentResult (sensor_msgs::Image &image) |
void | getCurrentSegmFlags (bool &with_colors, bool &with_color_holes_, bool &uniform_, bool &with_disparities_, bool &with_surface_) |
void | getCurrentSurface (float &alpha, float &beta, float &gamma) |
ObjectSegmenter (float grad_weight, int n_iter, bool use_gpu, float w_size=0.20f, float b_size=0.20f) | |
void | queueAction (const Action &action) |
void | setCurrentSegmFlags (bool with_colors, bool with_color_holes, bool uniform, bool with_disparities, bool with_surface) |
void | setNewData (const sensor_msgs::Image &image_, const stereo_msgs::DisparityImage &disparity_) |
~ObjectSegmenter () | |
Private Member Functions | |
void | clearQueue () |
bool | dequeueAction (Action &action) |
void | doSegment () |
void | fillImage (sensor_msgs::Image &image, const Image< uint8_t > &segm_mask) |
void | fillInitMask (Image< uint8_t > &init_mask, int start_x, int start_y, int stop_x, int stop_y) |
void | pause () |
void | setCurrentResult () |
template<typename Callable > | |
void | startThread (Callable f) |
void | stopThread () |
bool | validResult (const Image< uint8_t > &segm_mask) |
void | waitForAction () |
Private Attributes | |
std::deque< Action > | action_queue_ |
float | alpha_ |
float | ball_size_ |
float | beta_ |
boost::condition_variable | cond_var_ |
int | drange_ |
FgBgSegment * | fgbgsegment |
float | gamma_ |
Image< float > | gpu_disparities |
Image< uint8_t > | gpu_image |
int | grad_weight_ |
int | h_ |
boost::mutex | image_mutex_ |
bool | init_ |
Image< uint8_t > | init_mask |
int | n_iter_ |
int | num_fg_hypos_ |
boost::mutex | queue_mutex_ |
Image< uint8_t > | segm_mask |
boost::thread * | thread_ |
bool | use_gpu_ |
int | w_ |
float | window_size_ |
Definition at line 50 of file object_segmenter.h.
Definition at line 54 of file object_segmenter.h.
object_segmentation_gui::ObjectSegmenter::ObjectSegmenter | ( | float | grad_weight, |
int | n_iter, | ||
bool | use_gpu, | ||
float | w_size = 0.20f , |
||
float | b_size = 0.20f |
||
) |
Definition at line 45 of file object_segmenter.cpp.
Definition at line 65 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::clearQueue | ( | ) | [private] |
Definition at line 167 of file object_segmenter.cpp.
bool object_segmentation_gui::ObjectSegmenter::dequeueAction | ( | Action & | action | ) | [private] |
Definition at line 152 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::doSegment | ( | ) | [private] |
Definition at line 277 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::fillImage | ( | sensor_msgs::Image & | image, |
const Image< uint8_t > & | segm_mask | ||
) | [private] |
Definition at line 378 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::fillInitMask | ( | Image< uint8_t > & | init_mask, |
int | start_x, | ||
int | start_y, | ||
int | stop_x, | ||
int | stop_y | ||
) | [private] |
Definition at line 356 of file object_segmenter.cpp.
bool object_segmentation_gui::ObjectSegmenter::getCurrentResult | ( | sensor_msgs::Image & | image | ) |
Definition at line 192 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::getCurrentSegmFlags | ( | bool & | with_colors, |
bool & | with_color_holes_, | ||
bool & | uniform_, | ||
bool & | with_disparities_, | ||
bool & | with_surface_ | ||
) |
Definition at line 250 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::getCurrentSurface | ( | float & | alpha, |
float & | beta, | ||
float & | gamma | ||
) |
Definition at line 241 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::pause | ( | ) | [private] |
Definition at line 182 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::queueAction | ( | const Action & | action | ) |
Definition at line 141 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::setCurrentResult | ( | ) | [private] |
Definition at line 202 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::setCurrentSegmFlags | ( | bool | with_colors, |
bool | with_color_holes, | ||
bool | uniform, | ||
bool | with_disparities, | ||
bool | with_surface | ||
) |
Definition at line 263 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::setNewData | ( | const sensor_msgs::Image & | image_, |
const stereo_msgs::DisparityImage & | disparity_ | ||
) |
Definition at line 80 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::startThread | ( | Callable | f | ) | [private] |
Definition at line 339 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::stopThread | ( | ) | [private] |
Definition at line 348 of file object_segmenter.cpp.
bool object_segmentation_gui::ObjectSegmenter::validResult | ( | const Image< uint8_t > & | segm_mask | ) | [private] |
Definition at line 209 of file object_segmenter.cpp.
void object_segmentation_gui::ObjectSegmenter::waitForAction | ( | ) | [private] |
Definition at line 173 of file object_segmenter.cpp.
std::deque<Action> object_segmentation_gui::ObjectSegmenter::action_queue_ [private] |
Definition at line 163 of file object_segmenter.h.
float object_segmentation_gui::ObjectSegmenter::alpha_ [private] |
Definition at line 160 of file object_segmenter.h.
float object_segmentation_gui::ObjectSegmenter::ball_size_ [private] |
Definition at line 140 of file object_segmenter.h.
float object_segmentation_gui::ObjectSegmenter::beta_ [private] |
Definition at line 160 of file object_segmenter.h.
boost::condition_variable object_segmentation_gui::ObjectSegmenter::cond_var_ [private] |
Definition at line 169 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::drange_ [private] |
Definition at line 137 of file object_segmenter.h.
Definition at line 149 of file object_segmenter.h.
float object_segmentation_gui::ObjectSegmenter::gamma_ [private] |
Definition at line 160 of file object_segmenter.h.
Image<float> object_segmentation_gui::ObjectSegmenter::gpu_disparities [private] |
Definition at line 153 of file object_segmenter.h.
Image<uint8_t> object_segmentation_gui::ObjectSegmenter::gpu_image [private] |
Definition at line 152 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::grad_weight_ [private] |
Definition at line 136 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::h_ [private] |
Definition at line 135 of file object_segmenter.h.
boost::mutex object_segmentation_gui::ObjectSegmenter::image_mutex_ [private] |
Definition at line 167 of file object_segmenter.h.
Definition at line 145 of file object_segmenter.h.
Image<uint8_t> object_segmentation_gui::ObjectSegmenter::init_mask [private] |
Definition at line 156 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::n_iter_ [private] |
Definition at line 143 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::num_fg_hypos_ [private] |
Definition at line 147 of file object_segmenter.h.
boost::mutex object_segmentation_gui::ObjectSegmenter::queue_mutex_ [private] |
Definition at line 166 of file object_segmenter.h.
Image<uint8_t> object_segmentation_gui::ObjectSegmenter::segm_mask [private] |
Definition at line 159 of file object_segmenter.h.
boost::thread* object_segmentation_gui::ObjectSegmenter::thread_ [private] |
Definition at line 165 of file object_segmenter.h.
Definition at line 142 of file object_segmenter.h.
int object_segmentation_gui::ObjectSegmenter::w_ [private] |
Definition at line 134 of file object_segmenter.h.
float object_segmentation_gui::ObjectSegmenter::window_size_ [private] |
Definition at line 139 of file object_segmenter.h.