Store distance maps in pickle file

Enhance efficiency in training and testing by storing the distance maps within the pickle file.