Merge pull request #19 from antarctica/mg-imagery-bands-fix-0

Corrected bug for missing band

Author: EllieBowler

wfsai/__init__.py

@@ -4,4 +4,4 @@
 AI workflow stages of detecting Wildlife from Space.
 """
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"

wfsai/imagery.py

@@ -286,13 +286,14 @@ def orthorectify(self,
             gdal.UseExceptions()
             with gdal.Open(self.src) as dataset:
                 numbands = dataset.RasterCount
-                bandlist = [i for i in range(numbands)]
-                bandlist = list(set([1 if i == 0 else i for i in bandlist]))
+                bandlist = [i+1 for i in range(numbands)]
+                #bandlist = list(set([1 if i == 0 else i for i in bandlist]))
                 src_bands = dst_bands = bandlist
         elif src_bands is not None and dst_bands is None:
             dst_bands = src_bands
         else:
             pass
+        logger.info("dataset.RasterCount:          %s", str(numbands))
         logger.info("source bands:                 %s", str(src_bands))
         logger.info("destination bands:            %s", str(dst_bands))