Implement test-time augmentations for `rs predict`
We should implement optional support for test-time augmentation in rs predict.
Here is how it works: when predicting for a tile we will not only predict on the tile as is but in addition predict e.g. on the rotated and flipped versions of it. Then we would un-do the rotation or flipping on the masks, and merge these multiple predictions into one output.
Users can already do this by duplicating the slippy map directory with tiles to predict on, rotating or flipping them per slippy map directory. Then users need to run rs predict on all slippy map directories. And finally undo the transformations on the probability masks before using rs masks's support for model ensembles to get masks.
In contrast implementing test-time augmentation in rs predict needs to transform each tile on the fly (with our transformations) and then undo the transformations on the fly, too.
Tasks
- [ ] Implement optional switch in
rs predictfor test-time augmentations - [ ] Predict on tile and transformed tile in
rs predict - [ ] Undo transformations and merge predictions