// ------------------------------------------------------------------------
void computeProbImage(const PatchParams &params, unsigned int pnum,
                      unsigned int id, std::string type, ClassifierMap &classifiers,
                      const ValveType::Pointer &valve, const LabelType::Pointer &mask, RealImageType::Pointer &output)
{
    output = RealImageType::New();
    output->SetDirection(mask->GetDirection());
    output->SetSpacing(mask->GetSpacing());
    output->SetOrigin(mask->GetOrigin());
    output->SetRegions(mask->GetLargestPossibleRegion());
    output->Allocate();
    output->FillBuffer(0);

    itk::ImageRegionIterator<LabelType> maskIt(mask, mask->GetLargestPossibleRegion());
    itk::ImageRegionIterator<RealImageType> probIt(output, output->GetLargestPossibleRegion());

    unsigned int count = 0;
    while(!maskIt.IsAtEnd())
    {
        if(maskIt.Get() == 255)
        {
            IndexType index = maskIt.GetIndex();
            PointType point;
            output->TransformIndexToPhysicalPoint(index, point);

            MatrixType feature;
            extractLBPFeature(params, valve, point, feature);

            MatrixType probs;
            IntMatrixType classes;
            classifiers["MV-"+type][pnum]->PredictProbability(feature, classes, probs);

            probIt.Set(probs(0,1));

            count++;
        }


        ++maskIt;
        ++probIt;
    }



}
// ------------------------------------------------------------------------
void FlipImage(const LabelType::Pointer &input, LabelType::Pointer &output)
{
    if(!output) output = LabelType::New();

    ImageType::DirectionType outputDirection = input->GetDirection();

    typedef itk::PermuteAxesImageFilter<LabelType> FlipperType;
    FlipperType::PermuteOrderArrayType axes;
    axes[0] = 1;
    axes[1] = 0;
    axes[2] = 2;

    FlipperType::Pointer flipper = FlipperType::New();
    flipper->SetInput(input);
    flipper->SetOrder(axes);
    flipper->Update();

    output = flipper->GetOutput();
    output->SetDirection(outputDirection);

}