IBM / terratorch

a Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).
Apache License 2.0
151 stars 20 forks source link

Fixed transform in generic classification dataset #138

Closed blumenstiel closed 1 month ago

blumenstiel commented 2 months ago

The generic dataset resulted in an error because albumentations does not expect label and filename as an input.

blumenstiel commented 2 months ago

Also, _generate_bands_intervals was not present in the dataset.

CarlosGomes98 commented 1 month ago

@blumenstiel mostly added some cleanups piggy-backing on this PR. if you can check for your usecase as well and confirm it works we can merge

blumenstiel commented 1 month ago

@CarlosGomes98 @Joao-L-S-Almeida I added

warnings.filterwarnings("ignore", category=rasterio.errors.NotGeoreferencedWarning)

to the NotGeo datasets, as we don't need a geotransform for these datasets.