Method Load

src/colmap/mvs/workspace.cc:62–117

}

void Workspace::Load(const std::vector<std::string>& image_names) {
  const size_t num_images = model_.images.size();
  bitmaps_.resize(num_images);
  depth_maps_.resize(num_images);
  normal_maps_.resize(num_images);

  auto LoadWorkspaceData = [&, this](const int image_idx) {
    const size_t width = model_.images.at(image_idx).GetWidth();
    const size_t height = model_.images.at(image_idx).GetHeight();

    // Read and rescale bitmap
    auto bitmap = std::make_unique<Bitmap>();
    bitmap->Read(GetBitmapPath(image_idx), options_.image_as_rgb);
    if (bitmap->Width() != static_cast<int>(width) ||
        bitmap->Height() != static_cast<int>(height)) {
      bitmap->Rescale(static_cast<int>(width), static_cast<int>(height));
    }
    bitmaps_[image_idx] = std::move(bitmap);

    // Read and rescale depth map
    auto depth_map = std::make_unique<DepthMap>();
    depth_map->Read(GetDepthMapPath(image_idx));
    if (depth_map->GetWidth() != width || depth_map->GetHeight() != height) {
      depth_map->Downsize(width, height);
    }
    depth_maps_[image_idx] = std::move(depth_map);

    // Read and rescale normal map
    auto normal_map = std::make_unique<NormalMap>();
    normal_map->Read(GetNormalMapPath(image_idx));
    if (normal_map->GetWidth() != width || normal_map->GetHeight() != height) {
      normal_map->Downsize(width, height);
    }
    normal_maps_[image_idx] = std::move(normal_map);
  };

  const int num_threads = GetEffectiveNumThreads(options_.num_threads);
  ThreadPool thread_pool(num_threads);
  Timer timer;
  timer.Start();

  LOG(INFO) << StringPrintf("Loading workspace data with %d threads...",
                            num_threads);
  for (size_t i = 0; i < image_names.size(); ++i) {
    const int image_idx = model_.GetImageIdx(image_names[i]);
    if (HasBitmap(image_idx) && HasDepthMap(image_idx)) {
      thread_pool.AddTask(LoadWorkspaceData, image_idx);
    } else {
      LOG(WARNING) << StringPrintf(
          "Ignoring image %s, because input does not exist.",
          image_names[i].c_str());
    }
  }
  thread_pool.Wait();
  timer.PrintMinutes();
}

const Bitmap& Workspace::GetBitmap(const int image_idx) {