[WiP] Sen4Map inheritance class change to geobench-compatible datamodule and dataset classes

[MinasMayth]

Issue first raised in #226, Sen4map dataclass and datamodule classes currently only inherit from pytorch lightning, which is functionally adequate for terratorch but will not work with benchmark. I have updated the datamodule class to inherit from NonGeoDataModule, which required a few extra changes.

To get the Sen4map datamodule class to work with inheritance from NonGeoDataModule I had to define a custom augmentation self.aug = AugmentationSequential(K.RandomEqualize3D(), data_keys=None), is this okay? What is the default augmentation and why does it only (seemingly) work for BCHW tensors and not BCDHW tensors?

[Joao-L-S-Almeida]

Hi, @MinasMayth. It seems this work has entered in permanent regime. Do you feel confident to send it for review or prefer continue editing it ?
If so, you may add me as reviewer.

[MinasMayth]

HI, I am unable to continue working on this currently, so I would ask the issue be covered by someone else. Apologies! Kind regards, MinasMayth

…

On Wed, 26 Feb 2025 at 14:06, João Lucas de Sousa Almeida < ***@***.***> wrote: Hi, @MinasMayth <https://github.com/MinasMayth>. It seems this work has entered in permanent regime. Do you feel confident to send it for review or prefer continue editing it ? If so, you may add me as reviewer. — Reply to this email directly, view it on GitHub <#227 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AMYXIFERCVXZKTQXBAYQ3EL2RW34LAVCNFSM6AAAAABRLLYP22VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDMOBUHEYDKMZSHE> . You are receiving this because you were mentioned.Message ID: ***@***.***> [image: Joao-L-S-Almeida]*Joao-L-S-Almeida* left a comment (torchgeo/terratorch#227) <#227 (comment)> Hi, @MinasMayth <https://github.com/MinasMayth>. It seems this work has entered in permanent regime. Do you feel confident to send it for review or prefer continue editing it ? If so, you may add me as reviewer. — Reply to this email directly, view it on GitHub <#227 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AMYXIFERCVXZKTQXBAYQ3EL2RW34LAVCNFSM6AAAAABRLLYP22VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDMOBUHEYDKMZSHE> . You are receiving this because you were mentioned.Message ID: ***@***.***>

[Joao-L-S-Almeida]

Thanks @MinasMayth . I'll keep your branch open and I'll try to continue it.

[romeokienzler]

@MinasMayth @Joao-L-S-Almeida closing due to inactivity, feel free top reopen