{"cells":[{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","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)\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\nimport os\nprint(os.listdir(\"../input\"))\n\n# Any results you write to the current directory are saved as output.","execution_count":null,"outputs":[]},{"metadata":{"_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","trusted":true},"cell_type":"code","source":"import torch\nimport torchvision\nimport torch.nn as nn\nfrom torchvision.models import resnet152\nfrom torch.utils.data import DataLoader,Dataset\nfrom torch.utils.data.sampler import SubsetRandomSampler\nimport torchvision.transforms as transforms\nfrom PIL import Image\nimport torch.optim as optim","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df = pd.read_csv('../input/aptos2019-blindness-detection/train.csv')\nprint(df.shape)\ndf.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"class BlindDataset(Dataset):\n    def __init__(self,csv,transform):\n        self.data = pd.read_csv(csv)\n        self.transform = transform\n        self.labels = torch.eye(5)[self.data['diagnosis']]\n        \n    def __len__(self):\n        return len(self.data)\n    \n    def __getitem__(self,idx):\n        image_path = os.path.join('../input/aptos2019-blindness-detection/train_images/'+self.data.loc[idx]['id_code']+'.png')\n        image = Image.open(image_path)\n        image = self.transform(image)\n        label = torch.tensor(self.data.loc[idx]['diagnosis'])\n        return {'images':image,'labels':label}","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"simple_transform = transforms.Compose([transforms.Resize((224,224)),transforms.RandomHorizontalFlip(),transforms.RandomRotation(45),\n                                       transforms.RandomVerticalFlip(),transforms.RandomHorizontalFlip(),transforms.RandomRotation(45),transforms.ToTensor(),transforms.Normalize([0.496,0.456,0.406],[0.229,0.224,0.225])])","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_dataset = BlindDataset('../input/aptos2019-blindness-detection/train.csv',simple_transform)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"data_size = len(train_dataset)\nindices = list(range(data_size))\nsplit = int(np.round(0.1*data_size))\ntrain_indices = indices[split:]\nvalid_indics = indices[:split]","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_sampler = SubsetRandomSampler(train_indices)\nvalid_sampler = SubsetRandomSampler(valid_indics)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_loader = DataLoader(train_dataset,batch_size=32,sampler=train_sampler)\nvalid_loader = DataLoader(train_dataset,batch_size=32,sampler=valid_sampler)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"model = resnet152(pretrained=False)\nmodel.load_state_dict(torch.load('../input/resnet152/resnet152.pth'))\nfor param in model.parameters():\n    param.require_grad = False\n    \nmodel.fc = nn.Sequential(\nnn.Linear(2048,1024),\n    nn.Linear(1024,5))\nfc_parameters = model.fc.parameters()\nfor param in fc_parameters:\n    param.require_grad = True\nmodel = model.cuda()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"criteria = nn.CrossEntropyLoss()\noptimizer = optim.Adam(model.parameters(),lr=0.001)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def fit(epochs,model,optimizer,criteria):\n    for epoch in range(epochs+1):\n        training_loss = 0.0\n        validation_loss = 0.0\n        correct = 0.0\n        total = 0\n        print('{}/{} Epochs'.format(epoch+1,epochs))\n        \n        model.train()\n        for batch_idx,d in enumerate(train_loader):\n            data = d['images'].cuda()\n            target = d['labels'].cuda()\n            \n            optimizer.zero_grad()\n            output = model(data)\n            loss = criteria(output,target)\n            loss.backward()\n            optimizer.step()\n            \n            training_loss = training_loss + ((1/(batch_idx+1))*(loss.data-training_loss))\n            if batch_idx%20==0:\n                print('Training loss {}'.format(training_loss))\n            pred = output.data.max(1,keepdim=True)[1]\n            correct += np.sum(np.squeeze(pred.eq(target.data.view_as(pred))).cpu().numpy())\n            total +=data.size(0)\n            print('Accuracy on batch {} on Training is {}'.format(batch_idx,(100*correct/total)))\n            \n        model.eval()\n        for batch_idx ,d in enumerate(valid_loader):\n            data = d['images'].cuda()\n            target = d['labels'].cuda()\n            \n            output = model(data)\n            loss = criteria(output,target)\n            \n            validation_loss = validation_loss +((1/(batch_idx+1))*(loss.data-validation_loss))\n            if batch_idx%20==0:\n                print('Validation_loss {}'.format(validation_loss))\n            pred = output.data.max(1,keepdim=True)[1]\n            correct += np.sum(np.squeeze(pred.eq(target.data.view_as(pred))).cpu().numpy())\n            total+=data.size(0)\n            print('Validation Accuracy on Batch {} is {}'.format(batch_idx,(100*correct/total)))\n            \n    return model","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"fit(10,model,optimizer,criteria)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"class Prediction(Dataset):\n    def __init__(self,csv,transform):\n        self.data = pd.read_csv(csv)\n        self.transform = transform\n        \n    def __len__(self):\n        return len(self.data)\n    \n    def __getitem__(self,idx):\n        image_path = os.path.join('../input/aptos2019-blindness-detection/test_images/'+self.data.loc[idx]['id_code']+'.png')\n        image = Image.open(image_path)\n        image = self.transform(image)\n        return {'images':image}","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"test_dataset = Prediction('../input/aptos2019-blindness-detection/test.csv',simple_transform)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"test_loader = DataLoader(test_dataset)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"prediction = []\nfor batch_idx,d in enumerate(test_loader):\n    data = d['images'].cuda()\n    output = model(data)\n    output = output.cpu().detach().numpy()\n    prediction.append(np.argmax(output))","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sample_submission = pd.read_csv('../input/aptos2019-blindness-detection/sample_submission.csv')","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sample_submission.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sample_submission['diagnosis'] = prediction","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sample_submission.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sample_submission.to_csv('submission.csv',index = False)","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"name":"python","version":"3.6.4","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"}},"nbformat":4,"nbformat_minor":1}