Building a Crop Yield Prediction App Using Satellite Imagery and Jupyter

May 31, 2022
Reading Time
Rate this post
(19 votes)
Building a Crop Yield Prediction App Using Satellite Imagery and Jupyter

ACF Type: wysiwyg

A team of 30 AI engineers used Google Earth Engine (GEE) images and Jupyter to build an app for crop yield prediction in Senegal, Africa, to improve food security in the country.

Omdena´s Crop Yield Prediction AI Challenge in Africa

In this Omdena AI Challenge with the Global Partnership for Sustainable Development Data, we created a simple but powerful application using GEE images to estimate crop yield in Senegal.

For food security understanding the food system is essential. Accurate crop type details in near real-time will provide policymakers insights on the food system and will provide information on crop diversity and nutrition outcomes. So we created an application using open-source satellite images to identify the crops and estimate the yields for any given area.

Build your portfolio with real-world projects from Omdena

Challenges Encountered

1. The first challenge is to acquire satellite images from the regions selected by the user interactively over a map.

For this, we came up with the Python package Geemap which allows users to select a region in the map and get the geo-coordinates of the selected region.

!pip install geemap
import geemap
Map = geemap.Map()
Selecting a region from the map .  Source: Omdena

Selecting a region from the map.  Source: Omdena

#draw a rectangle in the above map and then run this
#ROI - Region of Interest
feature = Map.draw_last_feature
ROI = feature.geometry()


{'geodesic': False,
 'type': 'Polygon',
 'coordinates': [[[-120.021748, 46.126847],
   [-120.021748, 46.126957],
   [-120.021535, 46.126957],
   [-120.021535, 46.126847],
   [-120.021748, 46.126847]]]}

2. Downloading the geo tiff file and processing the file to get the data in the form of a matrix is computationally powerful.

We used the Geemap function ‘ee_to_numpy’ for this. It converts the image collected from the Google earth engine ‘ee.image’ into ‘NumPy’ array. It doesn’t need local computational power. It is all done in Google Earth Engine itself.

import ee
def maskS2clouds(image):
  qa ='QA60')
  cloudBitMask = 1 << 10
  cirrusBitMask = 1 << 11
  mask = qa.bitwiseAnd(cloudBitMask).eq(0).And(
  return image.updateMask(mask).divide(10000) 

# Load Sentinel-2 TOA reflectance data.
imageCollection = ee.ImageCollection('COPERNICUS/S2') \
    .filterDate('2016-01-01', '2016-12-31') \
    .filter('CLOUDY_PIXEL_PERCENTAGE', 5)) \
    .map(maskS2clouds) \
    .filterBounds(roi) \

rawImg = ee.Image(imageCollection.toList(imageCollection.size()).get(0))

#get the matrix for B2 band only
B2_image = geemap.ee_to_numpy(['B2']), region=ROI)

3. Each band in the sentinel is of a different resolution. So the matrix we got is of different shapes for each resolution. We wanted to have all the matrices stacked on top of each other to get a single multi-channel image.

Sentinel Dataset Band Info .  Source: Google Earth Engine Data Catalog

Sentinel Dataset Band Info.  Source: Google Earth Engine Data Catalog

We resized all the matrix to the high resolution (In this case 10 meters resolution) using the ‘nearest neighbor interpolation’ method.

targetSize = (50,50)
B5_Matrix = cv2.resize(B5_image,targetSize,interpolation=cv2.INTER_NEAREST)
Source: Omdena

Source: Omdena

Creating the Application

To create the application we used ‘ipywidgets’ which is an interactive HTML widget for Jupyter Notebook, Jupyter Lab, and Ipython Kernel. Then we used Voilà to turn the Jupyter notebooks into standalone web applications.

Step 1:  Define the user interface

Sample code

#Date picker 
dateBox = widgets.DatePicker(

mapWidget = widgets.Output()
step1 = widgets.Label('Step 1: Select the date')
step2 = widgets.Label('Step 2: Select Region from the map')
step3 = widgets.Label('Step 3: Load model')
step4 = widgets.Label('Step 4: Estimate yield')
estimate_yield_debug = widgets.Label('')
getROI = widgets.Button(description='Ok')
estimate_yield_Btn = widgets.Button(description='Estimate yield')
loadModel_Btn = widgets.Button(description='Load model')

#Progress bar
progressBar = widgets.IntProgress(

#Text Area
estimate_yield_Out = widgets.Textarea(
    layout={'height': '300px'}

#Display prediction image (Matplotlib plot)
estimate_yield_plot = widgets.Output()

Step 2:  Updating the widgets

1. Updating a label

for band in bands:
    estimate_yield_debug.value = f"Processing: {band}"
Source: Omdena

Source: Omdena

2. Reading information from date Picker

year  = dateBox.value.year
month = dateBox.value.month
day   =
Source: Omdena

Source: Omdena

3. To display Map

Map = geemap.Map(center=[14.607989,-14.531731], zoom=7)
Map.add_basemap('Google Satellite Hybrid')
with mapWidget:

4. To trigger a function when a button is clicked

def getStatistics(change):
      some codes

5. To update the progress bar

for band in bands:
      estimate_yield_debug.value = f"Processing: {band}"
      #Update progress bar
      progressBar.value = progressBar.value +1
Source: Omdena

Source: Omdena

6. To display matplot image

def plotResult(prediction):
    Code to visualize the prediction in the form of Matplotlib image.

    #display output
#Visualize the prediction
with estimate_yield_plot:
Using Jupyter Notebook to make a crop yield app. Source: Omdena

Using Jupyter Notebook to make a crop yield app. Source: Omdena

Step 3:  Create the app layout

Vertical Box (widgets.VBox) and Horizontal Box (widgets.HBox) are used to group widgets together.

Then we can use AppLayout from ipywidgets to align all the widgets in a proper way.

from ipywidgets import AppLayout
#Arrange the layout
verticalBox = mapWidget
vBox1 = widgets.VBox([step1, dateBox,vBox])
vBox2 = widgets.VBox([step2, getROI,
                      step3,loadModel_Btn,                            step4,estimate_yield_Btn,progressBar,estimate_yield_debug,yieldOutput,

          height="70%", width="150%")

Steps to run the application

1. Open the notebook

Using Jupyter Notebook to make a crop yield app. Source: Omdena

Using Jupyter Notebook to make a crop yield app. Source: Omdena

2. Click on the Voilà button

Voila button in Jupyter Notebook to make a crop yield app - Source: Omdena

Voila button in Jupyter Notebook to make a crop yield app – Source: Omdena

Check out the 2 mins app demo

Useful resources

ACF Type: image

Praveen Kanna

ACF Type: text

Praveen Kanna

ACF Type: url

Vetted Senior AI Talent

Work with our top 2% hidden gems, vetted through over 300 real-world projects.

Top Talent

Leave a comment.
  1. joseph

    Nice article!
    is there a github repo with the code and data you used?

  2. Issam Boukhris

    Thank you for this nice article.

    Is there any github repository where I can find the script you used ?

    With Thanks

    • Valentin

      Nice work! Congratulations.
      I am interested with the python script used to predict the yields, please.

      Thank you

  3. Rocco

    Can you provide the complete python code for this great app?

    Thank you

  4. Dishan Otieno

    Great work

Submit a Comment

Your email address will not be published. Required fields are marked *