I have an image directory (path_to_img_dir), where the image names have labels i.e. 1_1.png, 1_2.png .., 2_1.png, 2_2.png, etc. Here the first number before the _ is my labe