for f in img_files:
    num_subject = int(f.split("_")[1].split(".")[0])

    if num_subject < 311:
        shutil.copy(join(path_to_init_img, f), join(path_to_train, "images"))