Program design changed to support datasets with different ground truth and datasets with multiple images per label.