2edd1fe70a43b5b646662ef6603f579275141fae,examples/utils/tf_records_generation.py,,main,#,96

Before Change

    if not op.isdir(test_dir):
        makedirs(test_dir)

    for tile in list(tf_train[:,1]):
        tile_dir = op.join(tiles_dir, tile)
        shutil.copy(tile_dir, train_dir)

After Change

    column_name = ["filename", "width", "height", "class", "xmin", "ymin", "xmax", "ymax"]
    df = pd.DataFrame(tf_tiles_info, columns=column_name)
    //shuffle the dataframe
    df = df.sample(frac=1)
    train_df = df[:split_index]
    test_df = df[split_index:]
    print("You have {} training tiles and {} test tiles ready".format(
    len(set(train_df["filename"])), len(set(test_df["filename"]))))
    // train_df.to_csv("train_df.csv")

Instances