I just want to fix the focal length and distortion parameters K1, K2 of each camera. I don’t know how to rewrite it.
I used the follow method to directly fix all the parameters of a camera. Can you help me how to rewrite?
#include <cstdio>
#include <iostream>
#include<fstream>
#include "ceres/ceres.h"
#include "ceres/rotation.h"
#include "bal_problem.h"
// Read a Bundle Adjustment in the Large dataset.
class BALProblem {
public:
~BALProblem() {
delete[] point_index_;
delete[] camera_index_;
delete[] observations_;
delete[] parameters_;
}
int num_observations() const { return num_observations_; }
const double* observations() const { return observations_; }
double* mutable_cameras() { return parameters_; }
double* mutable_points() { return parameters_ + 9 * num_cameras_; }
double* mutable_camera_for_observation(int i) {
return mutable_cameras() + camera_index_[i] * 9;
}
double* mutable_point_for_observation(int i) {
return mutable_points() + point_index_[i] * 3;
}
bool LoadFile(const char* filename) {
FILE* fptr = fopen(filename, "r");
if (fptr == NULL) {
return false;
};
FscanfOrDie(fptr, "%d", &num_cameras_);
FscanfOrDie(fptr, "%d", &num_points_);
FscanfOrDie(fptr, "%d", &num_observations_);
point_index_ = new int[num_observations_];
camera_index_ = new int[num_observations_];
observations_ = new double[2 * num_observations_];
num_parameters_ = 9 * num_cameras_ + 3 * num_points_;
parameters_ = new double[num_parameters_];
parameters_1 = parameters_;
for (int i = 0; i < num_observations_; ++i) {
FscanfOrDie(fptr, "%d", camera_index_ + i);
FscanfOrDie(fptr, "%d", point_index_ + i);
for (int j = 0; j < 2; ++j) {
FscanfOrDie(fptr, "%lf", observations_ + 2 * i + j);
}
}
for (int i = 0; i < num_parameters_; ++i) {
FscanfOrDie(fptr, "%lf", parameters_ + i);
}
return true;
}
void point_print()
{
for (int j = 0; j < num_cameras_; j++)
{
std::cout << "optimization" << j << "camera information:" << "\n";
for (int i = 0 + j * 9; i < 9 + j * 9; i++)
{
std::cout << *(parameters_1 + i) << "\n";
}
}
for (int a = 0; a < num_points_; a++)
{
std::cout << "optimization" << a << "3Dpoint information:" << "\n";
for (int b = 9 * num_cameras_ + a * 3; b < 9 * num_cameras_ + 3 * (a + 1); b++)
{
std::cout << *(parameters_1 + b) << "\n";
}
}
}
private:
template<typename T>
void FscanfOrDie(FILE *fptr, const char *format, T *value) {
int num_scanned = fscanf(fptr, format, value);
if (num_scanned != 1) {
LOG(FATAL) << "Invalid UW data file.";
}
}
int num_cameras_;
int num_points_;
int num_observations_;
int num_parameters_;
int* point_index_;
int* camera_index_;
double* observations_;
double* parameters_;
double* parameters_1;
};
struct SnavelyReprojectionError {
SnavelyReprojectionError(double observed_x, double observed_y)
: observed_x(observed_x), observed_y(observed_y) {}
template <typename T>
bool operator()(const T* const camera,
const T* const point,
T* residuals) const {
T p[3];
ceres::AngleAxisRotatePoint(camera, point, p);
// camera[3,4,5] are the translation.
p[0] += camera[3];
p[1] += camera[4];
p[2] += camera[5];
//std::cout << "p[0]:" << p[0] << "\n" << "p[1]:" << p[1] << "\n" << "p[2]:" << p[2] << "\n";
T xp =- p[0] / p[2];
T yp =- p[1] / p[2];
//std::cout << "xp:" << xp << "\n" << "yp:" << yp << "\n" << "\n";
const T& l1 = camera[7];
const T& l2 = camera[8];
T r2 = xp * xp + yp * yp;
T distortion = 1.0 + r2 * (l1 + l2 * r2);
//std::cout << "r2:" << r2 << " " << "distortion:" << distortion << "\n" << "\n";
// Compute final projected point position.
const T& focal = camera[6];
T predicted_x = focal * distortion * xp;
T predicted_y = focal * distortion * yp;
//std::cout << "predicted_x:" << predicted_x << "\n" << "predicted_y:" << predicted_y <<"\n"<<"\n";
// The error is the difference between the predicted and observed position.
residuals[0] = predicted_x - observed_x;
residuals[1] = predicted_y - observed_y;
return true;
}
// Factory to hide the construction of the CostFunction object from
// the client code.
static ceres::CostFunction* Create(const double observed_x,
const double observed_y)
{
return (new ceres::AutoDiffCostFunction<SnavelyReprojectionError, 2, 9, 3>(
new SnavelyReprojectionError(observed_x, observed_y)));
}
double observed_x;
double observed_y;
};
int main(int argc, char** argv) {
google::InitGoogleLogging(argv[0]);
BALProblem bal_problem;
if (!bal_problem.LoadFile("C:\\02.txt")) {
std::cerr << "ERROR: unable to open file " << "\n";
return 1;
}
const double* observations = bal_problem.observations();
ceres::Problem problem;