Как использовать ceres ::valuation_callbacks для внутренней итерации функций ceres :: cost - PullRequest
0 голосов
/ 25 мая 2019

Я рассчитываю визуальное изображение на основе бумаги, а затем оптимизирую свои параметры: фокусное расстояние, вращение и перемещение.По этой причине я создаю функцию стоимости, перемещая все пиксели реального изображения и виртуального изображения.В своих функциях ceres cost я в основном вычитал нормализованное виртуальное изображение из нормализованного реального изображения.Виртуальный образ рассчитывается в функтореvaluation_callback, а стоимость - в функторе функции стоимости.Проблема связана с функтором стоимости.Оптимизация прекращается на первой итерации, потому что градиент равен 0. Я использую ceres :: Central для вычисления градиента, но функтор создания виртуальных изображений вызывается только один раз за каждую итерацию.Однако мне нужно, чтобы этот функтор вызывался для f (x + h) и f (xh) отдельно. Когда я вычисляю нормализованное реальное изображение и нормализованное виртуальное изображение по 9 соседям, я продолжаю итерацию, но каждая итерация занимает 25 секунд, что неприемлемо длямое дело.Мне нужна эта функцияvaluation_callback, но я не смог заставить ее работать.

Я смотрю на определениеvaluation_callbacks.записано, что «ПРИМЕЧАНИЕ: обратные вызовы оценки несовместимы с внутренними итерациями.»

struct RcpAndFpOptimizer {
    RcpAndFpOptimizer(cv::Mat &V,  const cv::Mat I, int i,int j,double width, double height) : V_(V), I_(I), i_(i),
                               j_(j), width_(width), height_(height){}
    bool operator()(const double* const fp, const double* const rotation, const double* const translation, double* residuals) const {


       double intensity = V_.at<double>(j_, i_);



        double tmp = (double)I_.at<double>(j_,i_)-(double)intensity;
        residuals[0] = tmp;
        //std::cout<<"pixels(i,j): "<<i_<<" "<<j_<<" residual: "<<residuals[0]<<std::endl;
        return true;
    }

    const cv::Mat S_;
    cv::Mat& V_;
    const cv::Mat I_;
    const int i_,j_;
    double width_, height_;
};


virtual void PrepareForEvaluation(bool evaluateJacobians, bool newEvaluationPoint)
    {
        if(evaluateJacobians){
            std::cout<<"evaluation jacobian is called"<<std::endl;
        }
        if (newEvaluationPoint)
        {
            // do your stuff here, e.g. calculate integral image

            //Mat V(height_, width_, CV_8UC1);

            std::cout<<"preperation is called"<<std::endl;


            Intrinsic<double> intrinsicC = INTRINSIC_CAMERA;
            Intrinsic<double> intrinsicP= {(double)fP_[0],(double)fP_[0], double(width_/2), double(height_/2), 0, 0};

            //Convertion of array to  point3d
            Point3d bDist = Point3d(translation_[0],translation_[1], translation_[2]);

            //Convertion euler array to rotation matrix
            const Mat eulerAngles = (cv::Mat_<double>(3,1) << rotArray_[0], rotArray_[1], rotArray_[2]);

            Mat rotM = rcpFinder::euler2rot(eulerAngles);

            Mat tempVImg(height_, width_, CV_8UC1);

            for (int i = 0; i < width_; ++i) {
                for (int j = 0; j < height_ ; ++j) {
                    //std::cout<<"Virtual current x and y pixels: "<<i<<" "<<j<<std::endl;
                    Point3d unprojPRay = rcpFinder::unprojectPoints(Point2i(i,j),intrinsicC);

                    //Assigning the intensity from images
                    tempVImg.at<uchar>(j, i)= rcpFinder::genVirtualImg(S_, intrinsicP, bDist, unprojPRay,
                                                            planeNormalAndDistance_,  rotM);

                    auto pixelIntensity = tempVImg.at<uchar>(Point(j, i));
                    //std::cout<<"pixel intensity "<< pixelIntensity<<std::endl;

                }

            }

            //imshow("Virtual", tempVImg);
            Mat integralV;
            cv::integral(tempVImg, integralV);

            //std::cout<<"integral image type is "<<integralV.type()<<std::endl;

            rcpFinder::normalizePixelsImg(tempVImg, integralV, V_);





            /*imshow("Normalized Img", V_);
            waitKey(0);*/

        }
    }

    // stuff here
    const cv::Mat S_;
    cv::Mat& V_;
    int width_, height_;
    map<int, vector<Point3d>> planeNormalAndDistance_;
    double *translation_;
    double* rotArray_;
    double* fP_;

};

// Вызов функторов подобен следующему

cv::Mat integralImgI;
    cv::integral(im1, integralImgI);
    cv::Mat normalizedRealImg;
    rcpFinder::normalizePixelsImg(im1, integralImgI, normalizedRealImg);

    Mat normalizedVirtualImg;

    //ceres::CostFunction* total_cost_function = 0;
    for (int i = 1; i < width-1; ++i) {
        for (int j = 1; j < height-1 ; ++j) {
            ceres::CostFunction* cost_function =
                    new ceres::NumericDiffCostFunction<RcpAndFpOptimizer, ceres::CENTRAL, 1, 1, 3, 3>(
                            new RcpAndFpOptimizer(normalizedVirtualImg, normalizedRealImg, i, j, width, height));

            problem.AddResidualBlock(cost_function, NULL, fp, rotationArray, translation);

        }
    }

    ceres::Solver::Options options;
    options.minimizer_progress_to_stdout = true;
    options.max_num_iterations = 50;
    options.update_state_every_iteration = true;



    options.evaluation_callback = (new evaluation_callback_functor(S, normalizedVirtualImg,width, height,
            mapNormalAndDist, translation,rotationArray, fp));

    ceres::Solver::Summary summary;
    ceres::Solve(options, &problem, &summary);

    std::cout << summary.BriefReport() << "\n";

Я ожидал, что решатель перестанет работать более чем одинпо крайней мере, итерация и градиент должны начинаться с некоторых значений и должны уменьшаться с помощью итерации.

Я нормализовал пизели с 9 соседями.Текущее решение, которое я нашел, вычисляет только 9 пикселей виртуального изображения в функторе затрат и использует их для нормализации одного пикселя, но это слишком медленно.У меня есть 640x480 пикселей и 9-кратный расчет для каждого пикселя.Плюс вычисление якобиана и градиента в NumericCOstFunction - это слишком много.Вот почему я хочу вычислить виртуальное изображение в функтореvaluation_callback и нормализовать его внутри этой функции и использовать нормализованное изображение в функторе затрат.Спасибо за вашу помощь.

1 Ответ

0 голосов
/ 05 июня 2019

Вы не можете вызвать оценочный колбэк с интератерациями

https://groups.google.com/forum/#!topic/ceres-solver/zjQLIaSuAdQ

...