示例#1
0
def write_features():
    with open(cfg.TRAIN_DATA_FILE, 'wb') as train_file:
        csv_writer = csv.writer(train_file)
        count = 0
        csv_writer.writerow(Features.get_header())
        for row in read_images():
            csv_writer.writerow(row)
            print count
            count += 1