{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"4769974a-6c1c-e3d3-1f30-52eff306fa39"},"outputs":[],"source":"# This Python 3 environment comes with many helpful analytics libraries installed\n# It is defined by the kaggle/python docker image: https://github.com/kaggle/docker-python\n# For example, here's several helpful packages to load in \n\nimport numpy as np # linear algebra\nimport pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)\nimport os\nimport matplotlib.pyplot as plt\nfrom PIL import Image\nplt.style.use('fivethirtyeight')\n\n# Input data files are available in the \"../input/\" directory.\n# For example, running this (by clicking run or pressing Shift+Enter) will list the files in the input directory\n\nfrom subprocess import check_output\nprint(check_output([\"ls\", \"../input\"]).decode(\"utf8\"))\n\n# Any results you write to the current directory are saved as output."},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"a50452ff-7ee5-5bc6-6efb-be9f236261a7"},"outputs":[],"source":"TEST_DIR = \"../input/test/\"\nTRAIN_DIR = \"../input/train/\""},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"7b6a037e-fb4a-5ebe-71df-0bcd174e1659"},"outputs":[],"source":"train_pic_dirs = os.listdir(TRAIN_DIR)\ntest_pic_dirs = os.listdir(TEST_DIR)\n\ndogs_train_dirs = [pic for pic in os.listdir(TRAIN_DIR) if 'dog' in pic]\ncats_train_dirs = [pic for pic in os.listdir(TRAIN_DIR) if 'cat' in pic]\n\nprint('Number of dogs in train path is {}'.format(len(dogs_train_dirs)))\nprint('Number of cats in train path is {}'.format(len(cats_train_dirs)))"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"29eab528-4cea-efb4-8723-c5d7155f8ba5"},"outputs":[],"source":"for pic in dogs_train_dirs[0:5]:\n    img = Image.open(TRAIN_DIR+pic)\n    img = np.array(img, 'f')\n    print(img.shape, img.dtype)\n    plt.imshow(img)"}],"metadata":{"_change_revision":0,"_is_fork":false,"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.5.2"}},"nbformat":4,"nbformat_minor":0}