[data.montgomery/shenzhen.loader] Simplify loaders, adjust some variable names, comments and add some (commented out) test code to visualize generated images

Maxime DELITROZ authored 1 year ago and André Anjos committed 1 year ago

We decided to handle the resizing of the input at the model level and not at the dataset level anymore. Loaders, docstrings and tests of Shenzhen and Montgomery were updated accordingly to only load data, remove black borders and crop to square shape. [image_utils.RemoveBlackBorders] had to be modified to allow functional utilization

[ptbench.data.datamodule] Implemented typing, added more logging, implemented sampler-balancing depending on sample class prevalence, parallelized dataset caching

[datamodule] Slightly streamlines the datamodule approach; adds documentation; adds type annotations; adds TODOs