f1fca244a7b6bff0ff66a052b58d3a3798d74ecb,imagenet-classification/create_input_files.py,,main,#Any#,20

Before Change

    paths = glob.glob("{}/*/*".format(args.train_file_dir))
    with open("train_label", "w") as fp:
        for path in paths:
            dname, fname = path.rstrip().split("/")[-2:]
            label = dname_label[dname]
            local_path = os.path.join(dname, fname)
            fp.write("{} {}\n".format(local_path, label))

After Change

            name_parts = path.parts[len(train_root.parts):]
            dname = name_parts[-1].split("_")[0]
            label = dname_label[dname]
            local_path = str(pathlib.Path(*name_parts))
            fp.write("{} {}\n".format(local_path, label))

    // File list for valdiation
    print("File list for valdiation")

Instances