Skip to content
Switch branches/tags
Go to file
Cannot retrieve contributors at this time
19 lines (16 sloc) 684 Bytes
from scipy.misc import imread
import numpy as np
import os
import pandas as pd
for directory, subdirectories, files in os.walk("./cat-train"): # for test: ./cat-test
for file in files:
if file == ".DS_Store": continue
im = imread(os.path.join(directory,file))
value = im.flatten()
value = np.hstack((directory[12:],value))
df = pd.DataFrame(value).T
with open('train_initial.csv', 'a') as f: #for test: test_initial.csv
df.to_csv(f, header=False, index=False)
df = pd.read_csv('train_initial.csv') #for test: test_initial.csv
df = df.sample(frac=1)
df.to_csv('train.csv',header=False, index=False) # for test: test.csv