teodorponcu / jigsaw-puzzle-transform Goto Github PK
View Code? Open in Web Editor NEWA repository that contains a simple function to get a patch permutation and its permutation indices. Contains a simple PyTorch dataloader wrapper as well.
License: MIT License