{"cells":[{"metadata":{},"cell_type":"markdown","source":"## To Know more about PyTorch you can check out this short [notes](https://github.com/shaft49/notes/tree/master/pytorch) that I've created while learning."},{"metadata":{},"cell_type":"markdown","source":"## Extract the data"},{"metadata":{"trusted":true},"cell_type":"code","source":"import shutil\nimport os\nfrom PIL import Image\ndef make_directory(dir_path):\n    if os.path.exists(dir_path):\n        shutil.rmtree(dir_path)\n    os.makedirs(dir_path)","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"import zipfile\n\n\nDATASET_PATH = '../output/dogs-vs-cats'\n\nmake_directory(DATASET_PATH)\n\n# extract train data\nwith zipfile.ZipFile('../input/dogs-vs-cats/train.zip', 'r') as zip_ref:\n    zip_ref.extractall(DATASET_PATH)\n\n# extract test data\nwith zipfile.ZipFile('../input/dogs-vs-cats/test1.zip', 'r') as zip_ref:\n    zip_ref.extractall(DATASET_PATH)\n","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","trusted":true},"cell_type":"code","source":"TRAIN_DATA_PATH = os.path.sep.join([DATASET_PATH, 'train'])\nTEST_DATA_PATH = os.path.sep.join([DATASET_PATH, 'test1'])\ntrain_files = os.listdir(TRAIN_DATA_PATH)\ntest_files = os.listdir(TEST_DATA_PATH)\n\nprint(len(train_files), len(test_files))","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Show Image"},{"metadata":{"trusted":true},"cell_type":"code","source":"Image.open(os.path.sep.join([TRAIN_DATA_PATH, train_files[0]])) ","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"Image.open(os.path.sep.join([TEST_DATA_PATH, test_files[0]])) ","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Import necessary libraries"},{"metadata":{"trusted":true},"cell_type":"code","source":"import torch\nimport torch.nn as nn\nimport torchvision\nfrom torch.utils.data import Dataset, DataLoader\nfrom torchvision import transforms","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Create Custom Dataset class and implement init, getitem & len function"},{"metadata":{"trusted":true},"cell_type":"code","source":"class CatsVsDogsDataset(Dataset):\n    def __init__(self):\n        pass\n    \n    def __getitem(self):\n        pass\n    \n    def __len__(self):\n        pass","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":""}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}