iately. 
              
              aug_img, aug_labels = PoseTools.preprocess_ims(img,labels,conf,scale=conf.rescale,distort=True)              
              # For PyTorch based networks, it is better to do the image augmentation in data_loader.
              # img and batches should be batched when calling preprocess_ims. The shape of img should be conf.batch_size x conf.imsz[0] x conf.imsz[1] x img_dim. For single animal, locs shape should be conf.batch_size x conf.n_classes x 2. For multi-animal, loc