Skip to content
Permalink
master
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