0

我正在开展一个使用数字地面模型 (DGM) 查找水深和范围的项目。我有多个覆盖感兴趣区域的 tiff 文件，我想将它们组合成一个 tiff 文件以进行快速处理。我如何使用下面我自己的代码或任何其他方法来组合它们？

我试图连接瓷砖，让它们一个一个地作为输入，然后将它们组合起来，但它GC error可能会抛出，因为代码本身有问题。下面提供了代码 

import geotrellis.proj4._
import geotrellis.raster._
import geotrellis.raster.io.geotiff._
object waterdepth {
  val directories = List("data")

  //constants to differentiate which bands to use
  val R_BAND = 0
  val G_BAND = 1
  val NIR_BAND = 2

  // Path to our landsat band geotiffs.
  def bandPath(directory: String) = s"../biggis-landuse/radar_data/${directory}"

  def main(args: Array[String]): Unit = {
    directories.map(directory => generateMultibandGeoTiffFile(directory))
  }

  def generateMultibandGeoTiffFile(directory: String) = {
    val tiffFiles = new java.io.File(bandPath(directory)).listFiles.map(_.toString)

    val singleBandGeoTiffArray = tiffFiles.foldLeft(Array[SinglebandGeoTiff]())((acc, el:String) => {
      acc :+ SinglebandGeoTiff(el)
    })

    val tileArray = ArrayMultibandTile(singleBandGeoTiffArray.map(_.tile))

    println(s"Writing out $directory multispectral tif")
    MultibandGeoTiff(tileArray, singleBandGeoTiffArray(0).extent, singleBandGeoTiffArray(0).crs).write(s"data/$directory.tif")

它应该能够从所有单独的文件中创建一个 tif 文件，但它会引发内存错误。

4

1 回答 1

1

您遵循的想法是正确的，可能会发生 OOM，因为您将大量 TIFF 加载到内存中，所以这并不奇怪。解决方案是为 JVM 分配更多内存。但是，您可以尝试这个小优化（可能会起作用）：

import geotrellis.proj4._
import geotrellis.raster._
import geotrellis.raster.io.geotiff._
import geotrellis.raster.io.geotiff.reader._

import java.io.File

def generateMultibandGeoTiffFile(directory: String) = {
  val tiffs =
    new File(bandPath(directory))
      .listFiles
      .map(_.toString)
      // streaming = true won't force all bytes to load into memory
      // only tiff metadata is fetched here
      .map(GeoTiffReader.readSingleband(_, streaming = true))

  val (extent, crs) = {
    val tiff = tiffs.head
    tiff.extent -> tiff.crs
  }

  // TIFF segments bytes fetch will start only during the write
  MultibandGeoTiff(
    MultibandTile(tiffs.map(_.tile)), 
    extent, crs
  ).write(s"data/$directory.tif")
}

}

于 2019-08-08T00:47:28.213 回答