Note that since xbatcher is made an optional dependency, users would need to do pip install zen3geo[raster] to install the extra 'raster' packages that includes xbatcher (and possibly more in the future).
TODO:
[x] Add xbatcher as optional dependency
[x] Initial implementation of XbatcherSlicerIterDataPipe
[x] Update CI build matrix to include optional xbatcher dependency in full tests run
[x] Add unit test for slicing xarray.Dataset objects
[x] Think about refactoring RioXarrayReader and PyogrioReader to not return tuples of (filename, dataobj) (maybe do in separate PR, edit: at #24)
An iterable-style DataPipe for creating chips from
xarray.DataArrayobjects! The windowed slicing is done using xbatcher.Note that
xbatcheris only used here to create slices of data (or chips/tiles), the actual creation of mini-batches (lumping together several slices/chips/tiles of images) will be handled by https://pytorch.org/data/0.4.0/generated/torchdata.datapipes.iter.Batcher.htmlPreview at https://zen3geo--22.org.readthedocs.build/en/19/api.html#module-zen3geo.datapipes.xbatcher
Note that since
xbatcheris made an optional dependency, users would need to dopip install zen3geo[raster]to install the extra 'raster' packages that includesxbatcher(and possibly more in the future).TODO:
xbatcheras optional dependencyXbatcherSlicerIterDataPipexbatcherdependency in full tests runxarray.Datasetobjects