In [22]:
%matplotlib inline
import numpy as np
import pandas as pd
import matplotlib
import matplotlib.pyplot as plt
plt.style.use('ggplot')
from os import getcwd
from os import listdir
from os.path import isfile, join, isdir
mypath = '/Users/jdstokes/repos/fish/'
trainPath = join(mypath,'train')
groupNames = [f for f in listdir(trainPath) if isdir(join(trainPath, f))]
In [9]:
import numpy as np
import pandas as pd
groupData = pd.DataFrame ({'group': groupNames})
# groupData.test = 'test'
In [10]:
# groupData.ix[0,'foo'] = 99
groupData
Out[10]:
In [131]:
fishPaths[groupData['group'][0]][0]
Out[131]:
In [143]:
fishPaths = dict.fromkeys(groupData['group'])
for idx,g in enumerate(groupNames):
fileNames = [f for f in listdir(join(trainPath,g)) if isfile(join(trainPath,g, f))]
for i,f in enumerate(fileNames):
fileNames[i] = join(trainPath,g,f)
groupData.ix[idx,'num files'] = int(len(fileNames))
fishPaths[g] = fileNames
groupData
Out[143]:
In [142]:
# groupData = groupData.drop('bar', 1)
In [137]:
import skimage
from skimage import io
camera = io.imread(fishPaths[groupData['group'][0]][0])
plt.figure(figsize=(5, 5))
plt.imshow(camera, cmap='gray', interpolation='nearest')
plt.axis('off')
plt.tight_layout()
plt.show()
print(camera.shape[0:2])
In [144]:
from skimage.color import rgb2gray
In [ ]: