for f in img_files:
    num_subject = int(f.split("_")[1].split(".")[0])
    if num_subject < 311:
        shutil.copy(join(path_to_init_label, f), join(path_to_train, "labels"))
    else: