A11c Checkpoint

Import libraries¶

In [ ]:

import ee
import geemap

import ee import geemap

Create an interactive map¶

In [ ]:

Map = geemap.Map(center=[40, -100], zoom=4)

Map = geemap.Map(center=[40, -100], zoom=4)

Add Earth Engine Python script¶

In [ ]:

# Add Earth Engine dataset
image = ee.Image("USGS/SRTMGL1_003")

#  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#  Chapter:      A1.1 Agricultural Environments
#  Checkpoint:   A11c
#  Authors:      Sherrie Wang, George Azzari
#  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

##############################
# 1. Pull all Landsat 7 and 8 images over study area
##############################

# Define study area.
TIGER = ee.FeatureCollection('TIGER/2018/Counties')
region = ee.Feature(TIGER \
    .filter(ee.Filter.eq('STATEFP', '17')) \
    .filter(ee.Filter.eq('NAME', 'McLean')) \
    .first())
geometry = region.geometry()
Map.centerObject(region)
Map.addLayer(region, {
    'color': 'red'
}, 'McLean County')

# Import Landsat imagery.
landsat7 = ee.ImageCollection('LANDSAT/LE07/C02/T1_L2')
landsat8 = ee.ImageCollection('LANDSAT/LC08/C02/T1_L2')

# Functions to rename Landsat 7 and 8 images.
def renameL7(img):
    return img.rename(['BLUE', 'GREEN', 'RED', 'NIR', 'SWIR1',
        'SWIR2', 'TEMP1', 'ATMOS_OPACITY', 'QA_CLOUD',
        'ATRAN', 'CDIST',
        'DRAD', 'EMIS', 'EMSD', 'QA', 'TRAD', 'URAD',
        'QA_PIXEL',
        'QA_RADSAT'
    ])


def renameL8(img):
    return img.rename(['AEROS', 'BLUE', 'GREEN', 'RED', 'NIR',
        'SWIR1',
        'SWIR2', 'TEMP1', 'QA_AEROSOL', 'ATRAN', 'CDIST',
        'DRAD', 'EMIS',
        'EMSD', 'QA', 'TRAD', 'URAD', 'QA_PIXEL', 'QA_RADSAT'
    ])


# Functions to mask out clouds, shadows, and other unwanted features.
def addMask(img):
    # Bit 0: Fill
    # Bit 1: Dilated Cloud
    # Bit 2: Cirrus (high confidence) (L8) or unused (L7)
    # Bit 3: Cloud
    # Bit 4: Cloud Shadow
    # Bit 5: Snow
    # Bit 6: Clear
    #        0: Cloud or Dilated Cloud bits are set
    #        1: Cloud and Dilated Cloud bits are not set
    # Bit 7: Water
    clear = img.select('QA_PIXEL').bitwiseAnd(64).neq(0)
    clear = clear.updateMask(clear).rename(['pxqa_clear'])

    water = img.select('QA_PIXEL').bitwiseAnd(128).neq(0)
    water = water.updateMask(water).rename(['pxqa_water'])

    cloud_shadow = img.select('QA_PIXEL').bitwiseAnd(16).neq(0)
    cloud_shadow = cloud_shadow.updateMask(cloud_shadow).rename([
        'pxqa_cloudshadow'
    ])

    snow = img.select('QA_PIXEL').bitwiseAnd(32).neq(0)
    snow = snow.updateMask(snow).rename(['pxqa_snow'])

    masks = ee.Image.cat([
        clear, water, cloud_shadow, snow
    ])

    return img.addBands(masks)


def maskQAClear(img):
    return img.updateMask(img.select('pxqa_clear'))


# Function to add GCVI as a band.
def addVIs(img):
  gcvi = img.expression('(nir / green) - 1', {
      'nir': img.select('NIR'),
      'green': img.select('GREEN')
  }).select([0], ['GCVI'])

  return ee.Image.cat([img, gcvi])


# Define study time period.
start_date = '2020-01-01'
end_date = '2020-12-31'

# Pull Landsat 7 and 8 imagery over the study area between start and end dates.
landsat7coll = landsat7 \
    .filterBounds(geometry) \
    .filterDate(start_date, end_date) \
    .map(renameL7)

landsat8coll = landsat8 \
    .filterDate(start_date, end_date) \
    .filterBounds(geometry) \
    .map(renameL8)

# Merge Landsat 7 and 8 collections.
landsat = landsat7coll.merge(landsat8coll) \
    .sort('system:time_start')

# Mask out non-clear pixels, add VIs and time variables.
landsat = landsat.map(addMask) \
    .map(maskQAClear) \
    .map(addVIs)

# Visualize GCVI time series at one location.
point = ee.Geometry.Point([-88.81417685576481,
    40.579804398254005
])
landsatChart = ui.Chart.image.series(landsat.select('GCVI'),
        point) \
    .setChartType('ScatterChart') \
    .setOptions({
        'title': 'Landsat GCVI time series',
        'lineWidth': 1,
        'pointSize': 3,
    })
print(landsatChart)

# Get crop type dataset.
cdl = ee.Image('USDA/NASS/CDL/2020').select(['cropland'])
Map.addLayer(cdl.clip(geometry), {}, 'CDL 2020')

#  -----------------------------------------------------------------------
#  CHECKPOINT
#  -----------------------------------------------------------------------

##############################
# 2. Add bands for harmonics
##############################

# Function that adds time band to an image.
def addTimeUnit(image, refdate):
    date = image.date()

    dyear = date.difference(refdate, 'year')
    t = image.select(0).multiply(0).add(dyear).select([0], ['t']) \
        .float()

    imageplus = image.addBands(t)

    return imageplus


# Function that adds harmonic basis to an image.
def addHarmonics(image, omega, refdate):
    image = addTimeUnit(image, refdate)
    timeRadians = image.select('t').multiply(2 * math.pi * omega)
    timeRadians2 = image.select('t').multiply(4 * math.pi *
    omega)

    return image \
        .addBands(timeRadians.cos().rename('cos')) \
        .addBands(timeRadians.sin().rename('sin')) \
        .addBands(timeRadians2.cos().rename('cos2')) \
        .addBands(timeRadians2.sin().rename('sin2')) \
        .addBands(timeRadians.divide(timeRadians) \
            .rename('constant'))


# Apply addHarmonics to Landsat image collection.
omega = 1
landsatPlus = landsat.map(
    def function(image):
        return addHarmonics(image, omega, start_date)
    )
print('Landsat collection with harmonic basis: ', landsatPlus)

#  -----------------------------------------------------------------------
#  CHECKPOINT
#  -----------------------------------------------------------------------

##############################
# 3. Get harmonic coefficients
##############################

# Function to run linear regression on an image.
def arrayimgHarmonicRegr(harmonicColl, dependent, independents):

    independents = ee.List(independents)
    dependent = ee.String(dependent)

    regression = harmonicColl \
        .select(independents.add(dependent)) \
        .reduce(ee.Reducer.linearRegression(independents.length(),
        1))

    return regression


# Function to extract and rename regression coefficients.
def imageHarmonicRegr(harmonicColl, dependent, independents):

    hregr = arrayimgHarmonicRegr(harmonicColl, dependent,
        independents)

    independents = ee.List(independents)
    dependent = ee.String(dependent)


def func_afr(b):
        return dependent.cat(ee.String('_')).cat(ee.String(
        b))

    newNames = independents.map(func_afr)





    imgCoeffs = hregr.select('coefficients') \
        .arrayProject([0]) \
        .arrayFlatten([independents]) \
        .select(independents, newNames)

    return imgCoeffs


# Function to apply imageHarmonicRegr and create a multi-band image.
def getHarmonicCoeffs(harmonicColl, bands, independents):
    coefficients = ee.ImageCollection.fromImages(bands.map(
        def function(band):
            return imageHarmonicRegr(harmonicColl, band,
                independents)
        ))
    return coefficients.toBands()


# Apply getHarmonicCoeffs to ImageCollection.
bands = ['NIR', 'SWIR1', 'SWIR2', 'GCVI']
independents = ee.List(['constant', 'cos', 'sin', 'cos2',
'sin2'])
harmonics = getHarmonicCoeffs(landsatPlus, bands, independents)

harmonics = harmonics.clip(geometry)
harmonics = harmonics.multiply(10000).toInt32()

# Compute fitted values.
gcviHarmonicCoefficients = harmonics \
    .select([
        '3_GCVI_constant', '3_GCVI_cos',
        '3_GCVI_sin', '3_GCVI_cos2', '3_GCVI_sin2'
    ]) \
    .divide(10000)


def func_gnr(image):
    return image.addBands(
        image.select(independents) \
        .multiply(gcviHarmonicCoefficients) \
        .reduce('sum') \
        .rename('fitted')
    )

fittedHarmonic = landsatPlus.map(func_gnr)









# Visualize the fitted harmonics in a chart.
harmonicsChart = ui.Chart.image.series(
        fittedHarmonic.select(
            ['fitted', 'GCVI']), point, ee.Reducer.mean(), 30) \
    .setSeriesNames(['GCVI', 'Fitted']) \
    .setOptions({
        'title': 'Landsat GCVI time series and fitted harmonic regression values',
        'lineWidth': 1,
        'pointSize': 3,
    })

print(harmonicsChart)

# Add CDL as a band to the harmonics.
harmonicsPlus = ee.Image.cat([harmonics, cdl])

# Export image to asset.
filename = 'McLean_County_harmonics'
Export.image.toAsset({
    'image': harmonicsPlus,
    'description': filename,
    'assetId': 'your_asset_path_here/' + filename,
    'dimensions': None,
    'region': region,
    'scale': 30,
    'maxPixels': 1e12
})


# Visualize harmonic coefficients on map.
visImage = ee.Image.cat([
    harmonics.select('3_GCVI_cos').divide(7000).add(0.6),
    harmonics.select('3_GCVI_sin').divide(7000).add(0.5),
    harmonics.select('3_GCVI_constant').divide(7000).subtract(
        0.6)
])

Map.addLayer(visImage, {
    'min': -0.5,
    'max': 0.5
}, 'Harmonic coefficient False color')

#  -----------------------------------------------------------------------
#  CHECKPOINT
#  -----------------------------------------------------------------------

# Add Earth Engine dataset image = ee.Image("USGS/SRTMGL1_003") # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ # Chapter: A1.1 Agricultural Environments # Checkpoint: A11c # Authors: Sherrie Wang, George Azzari # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ############################## # 1. Pull all Landsat 7 and 8 images over study area ############################## # Define study area. TIGER = ee.FeatureCollection('TIGER/2018/Counties') region = ee.Feature(TIGER \ .filter(ee.Filter.eq('STATEFP', '17')) \ .filter(ee.Filter.eq('NAME', 'McLean')) \ .first()) geometry = region.geometry() Map.centerObject(region) Map.addLayer(region, { 'color': 'red' }, 'McLean County') # Import Landsat imagery. landsat7 = ee.ImageCollection('LANDSAT/LE07/C02/T1_L2') landsat8 = ee.ImageCollection('LANDSAT/LC08/C02/T1_L2') # Functions to rename Landsat 7 and 8 images. def renameL7(img): return img.rename(['BLUE', 'GREEN', 'RED', 'NIR', 'SWIR1', 'SWIR2', 'TEMP1', 'ATMOS_OPACITY', 'QA_CLOUD', 'ATRAN', 'CDIST', 'DRAD', 'EMIS', 'EMSD', 'QA', 'TRAD', 'URAD', 'QA_PIXEL', 'QA_RADSAT' ]) def renameL8(img): return img.rename(['AEROS', 'BLUE', 'GREEN', 'RED', 'NIR', 'SWIR1', 'SWIR2', 'TEMP1', 'QA_AEROSOL', 'ATRAN', 'CDIST', 'DRAD', 'EMIS', 'EMSD', 'QA', 'TRAD', 'URAD', 'QA_PIXEL', 'QA_RADSAT' ]) # Functions to mask out clouds, shadows, and other unwanted features. def addMask(img): # Bit 0: Fill # Bit 1: Dilated Cloud # Bit 2: Cirrus (high confidence) (L8) or unused (L7) # Bit 3: Cloud # Bit 4: Cloud Shadow # Bit 5: Snow # Bit 6: Clear # 0: Cloud or Dilated Cloud bits are set # 1: Cloud and Dilated Cloud bits are not set # Bit 7: Water clear = img.select('QA_PIXEL').bitwiseAnd(64).neq(0) clear = clear.updateMask(clear).rename(['pxqa_clear']) water = img.select('QA_PIXEL').bitwiseAnd(128).neq(0) water = water.updateMask(water).rename(['pxqa_water']) cloud_shadow = img.select('QA_PIXEL').bitwiseAnd(16).neq(0) cloud_shadow = cloud_shadow.updateMask(cloud_shadow).rename([ 'pxqa_cloudshadow' ]) snow = img.select('QA_PIXEL').bitwiseAnd(32).neq(0) snow = snow.updateMask(snow).rename(['pxqa_snow']) masks = ee.Image.cat([ clear, water, cloud_shadow, snow ]) return img.addBands(masks) def maskQAClear(img): return img.updateMask(img.select('pxqa_clear')) # Function to add GCVI as a band. def addVIs(img): gcvi = img.expression('(nir / green) - 1', { 'nir': img.select('NIR'), 'green': img.select('GREEN') }).select([0], ['GCVI']) return ee.Image.cat([img, gcvi]) # Define study time period. start_date = '2020-01-01' end_date = '2020-12-31' # Pull Landsat 7 and 8 imagery over the study area between start and end dates. landsat7coll = landsat7 \ .filterBounds(geometry) \ .filterDate(start_date, end_date) \ .map(renameL7) landsat8coll = landsat8 \ .filterDate(start_date, end_date) \ .filterBounds(geometry) \ .map(renameL8) # Merge Landsat 7 and 8 collections. landsat = landsat7coll.merge(landsat8coll) \ .sort('system:time_start') # Mask out non-clear pixels, add VIs and time variables. landsat = landsat.map(addMask) \ .map(maskQAClear) \ .map(addVIs) # Visualize GCVI time series at one location. point = ee.Geometry.Point([-88.81417685576481, 40.579804398254005 ]) landsatChart = ui.Chart.image.series(landsat.select('GCVI'), point) \ .setChartType('ScatterChart') \ .setOptions({ 'title': 'Landsat GCVI time series', 'lineWidth': 1, 'pointSize': 3, }) print(landsatChart) # Get crop type dataset. cdl = ee.Image('USDA/NASS/CDL/2020').select(['cropland']) Map.addLayer(cdl.clip(geometry), {}, 'CDL 2020') # ----------------------------------------------------------------------- # CHECKPOINT # ----------------------------------------------------------------------- ############################## # 2. Add bands for harmonics ############################## # Function that adds time band to an image. def addTimeUnit(image, refdate): date = image.date() dyear = date.difference(refdate, 'year') t = image.select(0).multiply(0).add(dyear).select([0], ['t']) \ .float() imageplus = image.addBands(t) return imageplus # Function that adds harmonic basis to an image. def addHarmonics(image, omega, refdate): image = addTimeUnit(image, refdate) timeRadians = image.select('t').multiply(2 * math.pi * omega) timeRadians2 = image.select('t').multiply(4 * math.pi * omega) return image \ .addBands(timeRadians.cos().rename('cos')) \ .addBands(timeRadians.sin().rename('sin')) \ .addBands(timeRadians2.cos().rename('cos2')) \ .addBands(timeRadians2.sin().rename('sin2')) \ .addBands(timeRadians.divide(timeRadians) \ .rename('constant')) # Apply addHarmonics to Landsat image collection. omega = 1 landsatPlus = landsat.map( def function(image): return addHarmonics(image, omega, start_date) ) print('Landsat collection with harmonic basis: ', landsatPlus) # ----------------------------------------------------------------------- # CHECKPOINT # ----------------------------------------------------------------------- ############################## # 3. Get harmonic coefficients ############################## # Function to run linear regression on an image. def arrayimgHarmonicRegr(harmonicColl, dependent, independents): independents = ee.List(independents) dependent = ee.String(dependent) regression = harmonicColl \ .select(independents.add(dependent)) \ .reduce(ee.Reducer.linearRegression(independents.length(), 1)) return regression # Function to extract and rename regression coefficients. def imageHarmonicRegr(harmonicColl, dependent, independents): hregr = arrayimgHarmonicRegr(harmonicColl, dependent, independents) independents = ee.List(independents) dependent = ee.String(dependent) def func_afr(b): return dependent.cat(ee.String('_')).cat(ee.String( b)) newNames = independents.map(func_afr) imgCoeffs = hregr.select('coefficients') \ .arrayProject([0]) \ .arrayFlatten([independents]) \ .select(independents, newNames) return imgCoeffs # Function to apply imageHarmonicRegr and create a multi-band image. def getHarmonicCoeffs(harmonicColl, bands, independents): coefficients = ee.ImageCollection.fromImages(bands.map( def function(band): return imageHarmonicRegr(harmonicColl, band, independents) )) return coefficients.toBands() # Apply getHarmonicCoeffs to ImageCollection. bands = ['NIR', 'SWIR1', 'SWIR2', 'GCVI'] independents = ee.List(['constant', 'cos', 'sin', 'cos2', 'sin2']) harmonics = getHarmonicCoeffs(landsatPlus, bands, independents) harmonics = harmonics.clip(geometry) harmonics = harmonics.multiply(10000).toInt32() # Compute fitted values. gcviHarmonicCoefficients = harmonics \ .select([ '3_GCVI_constant', '3_GCVI_cos', '3_GCVI_sin', '3_GCVI_cos2', '3_GCVI_sin2' ]) \ .divide(10000) def func_gnr(image): return image.addBands( image.select(independents) \ .multiply(gcviHarmonicCoefficients) \ .reduce('sum') \ .rename('fitted') ) fittedHarmonic = landsatPlus.map(func_gnr) # Visualize the fitted harmonics in a chart. harmonicsChart = ui.Chart.image.series( fittedHarmonic.select( ['fitted', 'GCVI']), point, ee.Reducer.mean(), 30) \ .setSeriesNames(['GCVI', 'Fitted']) \ .setOptions({ 'title': 'Landsat GCVI time series and fitted harmonic regression values', 'lineWidth': 1, 'pointSize': 3, }) print(harmonicsChart) # Add CDL as a band to the harmonics. harmonicsPlus = ee.Image.cat([harmonics, cdl]) # Export image to asset. filename = 'McLean_County_harmonics' Export.image.toAsset({ 'image': harmonicsPlus, 'description': filename, 'assetId': 'your_asset_path_here/' + filename, 'dimensions': None, 'region': region, 'scale': 30, 'maxPixels': 1e12 }) # Visualize harmonic coefficients on map. visImage = ee.Image.cat([ harmonics.select('3_GCVI_cos').divide(7000).add(0.6), harmonics.select('3_GCVI_sin').divide(7000).add(0.5), harmonics.select('3_GCVI_constant').divide(7000).subtract( 0.6) ]) Map.addLayer(visImage, { 'min': -0.5, 'max': 0.5 }, 'Harmonic coefficient False color') # ----------------------------------------------------------------------- # CHECKPOINT # -----------------------------------------------------------------------

Display the interactive map¶

In [ ]:

Map

Map