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