JakubBodzioch commited on
Commit
f90ad6d
·
1 Parent(s): 9a592b8

Still testing NDBI

Browse files
io-app-backend/terramindFunctions.py CHANGED
@@ -29,7 +29,7 @@ NORMALIZATION_MODE = "offset"
29
  NDWI_THRESHOLD = 0.1
30
  MNDWI_THRESHOLD = 0.1
31
  NDVI_THRESHOLD = 0.3
32
- NDBI_THRESHOLD = 0.0
33
  BSI_THRESHOLD = 0.1
34
 
35
  USE_WATER_CORRECTION = True
@@ -340,7 +340,6 @@ def calculate_spectral_indices(input_tensor):
340
  green = input_tensor[0, 2].cpu().numpy() / 10000.0
341
  red = input_tensor[0, 3].cpu().numpy() / 10000.0
342
  nir = input_tensor[0, 7].cpu().numpy() / 10000.0
343
- nir_re = input_tensor[0, 8].cpu().numpy() / 10000.0
344
  swir1 = input_tensor[0, 10].cpu().numpy() / 10000.0
345
  swir2 = input_tensor[0, 11].cpu().numpy() / 10000.0
346
 
@@ -351,7 +350,7 @@ def calculate_spectral_indices(input_tensor):
351
  indices['awei'] = 4 * (green - swir1) - (0.25 * nir + 2.75 * swir2)
352
  indices['ndvi'] = (nir - red) / (nir + red + eps)
353
  indices['evi'] = 2.5 * (nir - red) / (nir + 6 * red - 7.5 * blue + 1 + eps)
354
- indices['ndbi'] = (swir1 - nir_re) / (swir1 + nir_re + eps)
355
  indices['bsi'] = ((swir1 + red) - (nir + blue)) / ((swir1 + red) + (nir + blue) + eps)
356
 
357
  return indices
 
29
  NDWI_THRESHOLD = 0.1
30
  MNDWI_THRESHOLD = 0.1
31
  NDVI_THRESHOLD = 0.3
32
+ NDBI_THRESHOLD = 0.08
33
  BSI_THRESHOLD = 0.1
34
 
35
  USE_WATER_CORRECTION = True
 
340
  green = input_tensor[0, 2].cpu().numpy() / 10000.0
341
  red = input_tensor[0, 3].cpu().numpy() / 10000.0
342
  nir = input_tensor[0, 7].cpu().numpy() / 10000.0
 
343
  swir1 = input_tensor[0, 10].cpu().numpy() / 10000.0
344
  swir2 = input_tensor[0, 11].cpu().numpy() / 10000.0
345
 
 
350
  indices['awei'] = 4 * (green - swir1) - (0.25 * nir + 2.75 * swir2)
351
  indices['ndvi'] = (nir - red) / (nir + red + eps)
352
  indices['evi'] = 2.5 * (nir - red) / (nir + 6 * red - 7.5 * blue + 1 + eps)
353
+ indices['ndbi'] = (swir1 - swir2) / (swir1 + swir2 + eps)
354
  indices['bsi'] = ((swir1 + red) - (nir + blue)) / ((swir1 + red) + (nir + blue) + eps)
355
 
356
  return indices