add band number and band indices as optional config arguments

Author: martham93

docs/parameters.rst

@@ -72,3 +72,10 @@ Here is the full list of configuration parameters you can specify in a ``config.
 
 **over_zoom**: int
 	An integer greater than 0. If set for XYZ tiles, it will fetch tiles from `zoom` + `over_zoom`, to create higher resolution tiles which fill out the bounds of the original zoom level.
+
+**band_count**: int 
+An integret greater than 1. If not specified defaults to 3 for the typical r,g,b imagery. 
+
+**band_indicies**: tuple 
+A tuple of band incidies to pull from a tiff. Defaults to (1, 2, 3) if not specified. For using non-RGB tifs, ie MS tifs this should be over-written 
+for example to (5, 3, 2, 7) to extract Red, Green, Blue, and NIR bands from the MS spacenet imagery. 

label_maker/utils.py

@@ -137,8 +137,21 @@ def get_tile_tif(tile, imagery, folder, kwargs):
  window = ((top, bottom), (left, right))
 
  # read the first three bands (assumed RGB) of the TIF into an array
- data = np.empty(shape=(3, 256, 256)).astype(src.profile['dtype'])
- for k in (1, 2, 3):
+
+ try:
+ kwargs['band_count']
+ band_count = kwargs['band_count']
+ except KeyError:
+ band_count = 3
+ 
+ try:
+ kwargs['band_indicies']
+ band_count = kwargs['band_indicies']
+ except KeyError:
+ band_indicies = (1, 2, 3)
+
+ data = np.empty(shape=(band_count, 256, 256)).astype(src.profile['dtype'])
+ for k in band_indicies:
  src.read(k, window=window, out=data[k - 1], boundless=True)
 
  # save

label_maker/validate.py

@@ -35,5 +35,7 @@
  'split_vals': {'type': 'list', 'schema': {'type': 'float'}},
  'split_names': {'type': 'list', 'schema': {'type': 'string'}},
  'tms_image_format': {'type': 'string'},
- 'over_zoom': {'type': 'integer', 'min': 1}
+ 'over_zoom': {'type': 'integer', 'min': 1}, 
+ 'band_count': {'type': 'integer', 'min': 1},
+ 'band_indicies': {'type': 'tuple'}
 }