将图像数组导出到Google Earth Engine中的TFRecord

时间:2019-06-20 18:32:42

标签: export google-earth-engine

我想将3个Landsat图像(每个12个波段)的集合转换为单个图像数组,然后以TFRecord格式导出。我用下面的代码。我的输入集合名为imagesimageT是累积的图像,从零带开始,最后被丢弃。最终imageOfSeries图像的每个像素都包含一个3x12大小的矩阵:

var imageT = ee.Image(0)
images = images.map(function(image){
  return image.toArray();
})   
var accumulate = function(image, imageT) {
  return(ee.Image(imageT).addBands(image)) 
};
var imageOfSeries = ee.Image(images.iterate(accumulate, imageT))
                             .slice(1).toArray(1).matrixTranspose()        
Export.image.toDrive({
  image: imageOfSeries,
  description: 'imageOfSeriesExample',
  scale: 30,
  region: geometry,
  fileFormat: 'TFRecord',
  formatOptions: {
    patchDimensions: [10,10],
    tensorDepths: [3,12]
  }
});

但是GEE在运行导出任务时返回错误,并说Arrays must have dimensions = 1。 我该怎么做?在这种情况下,我还需要有关如何解码TFRecord文件的更多信息,对此我在GEE教程中找不到任何示例。

1 个答案:

答案 0 :(得分:2)

要执行所需的操作,将需要展平矩阵以进行导出:导出到TFRecord仅支持一维数组(错误指出)。目前,EE不支持直接数组展平,而是希望为数组中每个展平的元素添加标签。

张量深度还需要按顺序排列数组带的一维长度。

假设您的图片正好有一个数组带,这应该可以工作(详细说明):

var imageT = ee.Image(0)
images = images.map(function(image){
  return image.toArray();
})   
var accumulate = function(image, imageT) {
  return(ee.Image(imageT).addBands(image)) 
};
var imageOfSeries = ee.Image(images.iterate(accumulate, imageT))
                             .slice(1).toArray(1).matrixTranspose()  

imageOfSeriesFlattened = imageOfSeries
    .arrayFlatten([
        ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12'],
        ['13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24'],
        ['25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36']])
    .toArray();  

Export.image.toDrive({
  image: imageOfSeriesFlattened,
  description: 'imageOfSeriesExample',
  scale: 30,
  region: geometry,
  fileFormat: 'TFRecord',
  formatOptions: {
    patchDimensions: [10,10],
    tensorDepths: [36]
  }
});