Update xView dataset to return image1, image2

microsoft / torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

https://www.osgeo.org/projects/torchgeo/

MIT License

2.68k stars 329 forks source link

Update xView dataset to return image1, image2 #1691

Open robmarkcole opened 11 months ago

robmarkcole commented 11 months ago

Summary

https://github.com/microsoft/torchgeo/pull/1684 updated OSCD to return ["image1", "image2", "mask"]. This issue is to track other CD datasets that require this change:

LEVIRCDPlus - DONE
xview2 - requires change to AugmentationSequential to handle [mask1, mask2]

Rationale

Make consistent interface for change detection models

Implementation

No response

Alternatives

No response

Additional information

No response

adamjstewart commented 11 months ago

[!NOTE]
Once we have a ChangeDetectionTask, we can remove the tests/datamodules/*.py files for these datasets and test them properly with a real trainer. Related to #978.

adamjstewart commented 11 months ago

Are there any change detection datasets with more than 2 images? Wonder if it would make sense to model this similar to a time-series dataset. @nilsleh is working on this but idk if we've finalized the design yet.

robmarkcole commented 11 months ago

Possibly Spacenet 7 - but broadly speaking change detection is a special case of time series, so an API that extends image1, image2, 3,4 etc makes sense