If one wants to obtain label maps for an entire dataset, one would want to maximize network throughput.
We provide the option here to perform preprocessing in the dataset and replace OpenCV calls with native pytorch interpolation functions to speed up everything significantly.