utils.py 文件源码

def prepare_data(imagery_path, train_file_path, split_points):

    # Read tiff image
    image_tuple = read_image(imagery_path)

    # Read samples
    original_data_list = []
    csv_reader = csv.reader(open(train_file_path, encoding='utf-8'))
    for row in csv_reader:
        original_data_list.append(row)
    original_data_array = np.array(original_data_list)

    # Split training data into variables and lables 
    x_s = original_data_array[:,split_points[0]:split_points[1]] 
    y_s = original_data_array[:,split_points[1]]

    return x_s, y_s, image_tuple


# Read image