TrainingViews.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
 
#include <Eigen/Core>
#include <opencv2/core.hpp>
 
#include <cstddef>
#include <string>
#include <vector>
 
namespace reusex {
class ProjectDB;
}
 
namespace reusex::gsplat {

struct TrainingView {
  int id = -1;       
  std::string name;  
  cv::Mat image;     
  Eigen::Matrix3d K; 
  Eigen::Matrix4d T_cw = Eigen::Matrix4d::Identity(); 
  bool from_panorama = false; 
 
  int width() const { return image.cols; }
  int height() const { return image.rows; }
};

struct TrainingViewOptions {
  bool include_frames = true;

  bool include_panorama_slices = false;
  int pano_n_yaw = 8;         
  double pano_fov_deg = 90.0; 
  int pano_tile = 1024;       

  int frame_stride = 1;

  int first_frame = -1;
  int last_frame = -1;

  int max_image_size = 0;

  std::size_t max_views = 0;
};

std::vector<TrainingView>
load_training_views(const ProjectDB &db, const TrainingViewOptions &opt = {});
 
} // namespace reusex::gsplat