Function CreateDatabaseWithRandomDescriptors

src/colmap/scene/database_test.cc:820–842

Helper to create a database file with images and descriptors.

// Helper to create a database file with images and descriptors.
std::shared_ptr<Database> CreateDatabaseWithRandomDescriptors(
    const std::vector<int>& num_descriptors_per_image) {
  auto database = Database::Open(kInMemorySqliteDatabasePath);

  const int num_images = num_descriptors_per_image.size();

  Camera camera;
  camera.camera_id = database->WriteCamera(camera);

  for (int i = 0; i < num_images; ++i) {
    Image image;
    image.SetName("image" + std::to_string(i));
    image.SetCameraId(camera.camera_id);
    image.SetImageId(database->WriteImage(image));
    database->WriteDescriptors(
        image.ImageId(),
        FeatureDescriptors(
            FeatureExtractorType::SIFT,
            FeatureDescriptorsData::Random(num_descriptors_per_image[i], 128)));
  }

  return database;
}

TEST(LoadRandomDatabaseDescriptorsTest, LoadEmpty) {
  auto database = Database::Open(kInMemorySqliteDatabasePath);