{"cells":[{"metadata":{"_uuid":"1d47ba4d-6a22-4f44-92b7-54c348a14b90","_cell_guid":"869d9bc5-7ade-4ead-8c77-894296cd87da","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 all files under the input directory\n\nimport os\n# for dirname, _, filenames in os.walk('/kaggle/input'):\n#     for filename in filenames:\n#         print(os.path.join(dirname, filename))\nprint(os.listdir(\"../input\"))\n\n# Any results you write to the current directory are saved as output.","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"import torch\nimport torchvision\nimport torch.nn as nn\nfrom torchvision.models import resnet101\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\nfrom torch.optim import lr_scheduler","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 EyeDataset(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":"data_transform = transforms.Compose([ transforms.RandomResizedCrop((224,224)),\n                                     transforms.RandomHorizontalFlip(),\n                                     transforms.ToTensor(),\n                                     transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_dataset = EyeDataset('../input/aptos2019-blindness-detection/train.csv',data_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]\ntrain_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 = resnet101(pretrained=False)\nmodel.load_state_dict(torch.load('../input/resnet101/resnet101.pth'))\nfor param in model.parameters():\n    param.require_grad = False\n    \n    \nmodel.avg_pool = nn.AdaptiveAvgPool2d(1)\nmodel.fc = nn.Sequential(\n              nn.BatchNorm1d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True),\n              nn.Dropout(p=0.25),\n              nn.Linear(in_features=2048, out_features=2048, bias=True),\n              nn.ReLU(),\n              nn.BatchNorm1d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True),\n              nn.Dropout(p=0.5),\n              nn.Linear(in_features=2048, out_features=5, bias=True),\n             )\nfc_parameters = model.fc.parameters()\nfor param in fc_parameters:\n    param.require_grad = True\nmodel = model.cuda()\ndevice = torch.device(\"cuda:0\")","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"criteria = nn.CrossEntropyLoss()\nplist = [\n        {'params': model.layer4.parameters(), 'lr': 1e-5}\n        ]\noptimizer_ft = optim.Adam(plist, lr=0.001)\nlr_sch = lr_scheduler.StepLR(optimizer_ft, step_size=10, gamma=0.1)\n","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def fit(epochs,model,optimizer,criteria, scheduler):\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_ft,criteria, lr_sch)","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',data_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":"submission = pd.read_csv('../input/aptos2019-blindness-detection/sample_submission.csv')\nsubmission.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"submission['diagnosis'] = prediction\nsubmission.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"submission.to_csv('submission.csv',index = False)","execution_count":null,"outputs":[]}],"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":1}