box

geosentinel/ndvi.py

@@ -5,6 +5,8 @@
 import sys
 import numpy as np
 import gdal
+from osgeo import osr
+
 def geoint_pr_ndvi(nir_fn, vir_fn, outfile):
 	'''!
 		Normalized Difference Vegetation Index (NDVI)
@@ -43,6 +45,8 @@ def geoint_pr_ndvi(nir_fn, vir_fn, outfile):
 	output.GetRasterBand(1).WriteArray(result)
 	output.GetRasterBand(1).FlushCache()
 	output.SetGeoTransform(geotransform)
+	wkt = nir.GetProjection()
+	output.SetProjection(osr.SpatialReference().ImportFromWkt(wkt).ExportToWkt())
 	output = None