{"cells":[{"metadata":{},"cell_type":"markdown","source":"Training kernel: https://www.kaggle.com/ateplyuk/panda-tpu-starter-train"},{"metadata":{"trusted":false},"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nimport os\nimport sys\nfrom tensorflow.keras.models import load_model\nimport cv2\nimport skimage.io\n\nsys.path.insert(0, '/kaggle/input/efficientnet-keras-source-code/')\nimport efficientnet.tfkeras as efn","execution_count":null,"outputs":[]},{"metadata":{"trusted":false},"cell_type":"code","source":"img_size = 512\ntest_dir = '/kaggle/input/prostate-cancer-grade-assessment/test_images'\nmodel = load_model('/kaggle/input/panda-tpu-starter-train/model.h5')","execution_count":null,"outputs":[]},{"metadata":{"trusted":false},"cell_type":"code","source":"def get_image(img_name):\n    data_dir = test_dir\n    img_path = os.path.join(data_dir, f'{img_name}.tiff')    \n    img = skimage.io.MultiImage(img_path)        \n    img = cv2.resize(img[-1], (img_size, img_size))    \n    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)\n    img = img / 255\n    return img","execution_count":null,"outputs":[]},{"metadata":{"trusted":false},"cell_type":"code","source":"sub = pd.read_csv('/kaggle/input/prostate-cancer-grade-assessment/sample_submission.csv')\ntest_images = sub['image_id'].values","execution_count":null,"outputs":[]},{"metadata":{"trusted":false},"cell_type":"code","source":"labels = []\ntry:\n    for image in test_images:                \n        img = get_image(image)    \n        im1 = img.reshape((1, img_size, img_size, 3))    \n        preds = model.predict(im1, batch_size=1)    \n        result = np.argmax(preds ,axis = 1)               \n        labels.append(result)\n    sub['isup_grade'] = labels\nexcept:\n    print('exception')","execution_count":null,"outputs":[]},{"metadata":{"trusted":false},"cell_type":"code","source":"sub['isup_grade'] = sub['isup_grade'].astype(int)\nsub.to_csv('submission.csv', index=False)\nsub.head()","execution_count":null,"outputs":[]}],"metadata":{"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.6.6"}},"nbformat":4,"nbformat_minor":4}