{"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_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import numpy as np # linear algebra\nimport pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)\nimport os\nimport tensorflow as tf","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:49:30.946607Z","iopub.execute_input":"2022-01-02T11:49:30.946948Z","iopub.status.idle":"2022-01-02T11:49:35.333960Z","shell.execute_reply.started":"2022-01-02T11:49:30.946910Z","shell.execute_reply":"2022-01-02T11:49:35.333164Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def auto_select_accelerator():\n    try:\n        tpu = tf.distribute.cluster_resolver.TPUClusterResolver()\n        tf.config.experimental_connect_to_cluster(tpu)\n        tf.tpu.experimental.initialize_tpu_system(tpu)\n        strategy = tf.distribute.experimental.TPUStrategy(tpu)\n        print(\"Running on TPU:\", tpu.master())\n    except ValueError:\n        strategy = tf.distribute.get_strategy()\n    print(f\"Running on {strategy.num_replicas_in_sync} replicas\")\n    \n    return strategy\n\n\ndef build_decoder(with_labels=True, target_size=(380, 380), ext='jpg'):\n    def decode(path):\n        file_bytes = tf.io.read_file(path)\n        if ext == 'png':\n            img = tf.image.decode_png(file_bytes, channels=3)\n        elif ext in ['jpg', 'jpeg']:\n            img = tf.image.decode_jpeg(file_bytes, channels=3)\n        else:\n            raise ValueError(\"Image extension not supported\")\n\n        img = tf.cast(img, tf.float32) / 255.0\n        img = tf.image.resize(img, target_size)\n\n        return img\n    \n    def decode_with_labels(path, label):\n        return decode(path), label\n    \n    return decode_with_labels if with_labels else decode\n\n\ndef build_augmenter(with_labels=True):\n    def augment(img):\n        #img = tf.image.random_flip_left_right(img)\n        #img = tf.image.random_flip_up_down(img)\n        return img\n    \n    def augment_with_labels(img, label):\n        return augment(img), label\n    \n    return augment_with_labels if with_labels else augment\n\n\ndef build_dataset(paths, labels=None, bsize=64, cache=True,\n                  decode_fn=None, augment_fn=None,\n                  augment=True, repeat=True, shuffle=1024, \n                  cache_dir=\"\"):\n    if cache_dir != \"\" and cache is True:\n        os.makedirs(cache_dir, exist_ok=True)\n    \n    if decode_fn is None:\n        decode_fn = build_decoder(labels is not None)\n    \n    if augment_fn is None:\n        augment_fn = build_augmenter(labels is not None)\n    \n    AUTO = tf.data.experimental.AUTOTUNE\n    slices = paths if labels is None else (paths, labels)\n    \n    dset = tf.data.Dataset.from_tensor_slices(slices)\n    dset = dset.map(decode_fn, num_parallel_calls=AUTO)\n    dset = dset.cache(cache_dir) if cache else dset\n    dset = dset.map(augment_fn, num_parallel_calls=AUTO) if augment else dset\n    dset = dset.repeat() if repeat else dset\n    dset = dset.shuffle(shuffle) if shuffle else dset\n    dset = dset.batch(bsize).prefetch(AUTO)\n    \n    return dset","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:49:40.455701Z","iopub.execute_input":"2022-01-02T11:49:40.456140Z","iopub.status.idle":"2022-01-02T11:49:40.490069Z","shell.execute_reply.started":"2022-01-02T11:49:40.456100Z","shell.execute_reply":"2022-01-02T11:49:40.489010Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"\n\nCOMPETITION_NAME = \"ranzcr-clip-catheter-line-classification\"\nstrategy = auto_select_accelerator()\nBATCH_SIZE = strategy.num_replicas_in_sync * 16\n\n","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:49:47.999521Z","iopub.execute_input":"2022-01-02T11:49:47.999852Z","iopub.status.idle":"2022-01-02T11:49:48.012601Z","shell.execute_reply.started":"2022-01-02T11:49:47.999823Z","shell.execute_reply":"2022-01-02T11:49:48.011591Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"IMSIZE = (224, 224, 260, 300, 380, 456, 528, 600)\n\nload_dir = f\"/kaggle/input/{COMPETITION_NAME}/\"\nsub_df = pd.read_csv(load_dir + 'sample_submission.csv')\ntest_paths = load_dir + \"test/\" + sub_df['StudyInstanceUID'] + '.jpg'\n\n# Get the multi-labels\nlabel_cols = sub_df.columns[1:]\n\ntest_decoder = build_decoder(with_labels=False, target_size=(IMSIZE[4], IMSIZE[4]))\ndtest = build_dataset(\n    test_paths, bsize=BATCH_SIZE, repeat=False, \n    shuffle=False, augment=False, cache=False,\n    decode_fn=test_decoder\n)","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:49:59.510003Z","iopub.execute_input":"2022-01-02T11:49:59.510374Z","iopub.status.idle":"2022-01-02T11:50:01.613027Z","shell.execute_reply.started":"2022-01-02T11:49:59.510341Z","shell.execute_reply":"2022-01-02T11:50:01.612282Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"\nmodel = tf.keras.models.load_model('../input/resnet/Exp_ResNetV2380_00.h5')","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:50:17.578185Z","iopub.execute_input":"2022-01-02T11:50:17.578519Z","iopub.status.idle":"2022-01-02T11:50:25.885628Z","shell.execute_reply.started":"2022-01-02T11:50:17.578488Z","shell.execute_reply":"2022-01-02T11:50:25.884794Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"\n\nsub_df[label_cols] = model.predict(dtest, verbose=1)\nsub_df.to_csv('submission.csv', index=False)\n\nsub_df.head()","metadata":{"execution":{"iopub.status.busy":"2022-01-02T11:50:28.361166Z","iopub.execute_input":"2022-01-02T11:50:28.361495Z","iopub.status.idle":"2022-01-02T11:54:30.057068Z","shell.execute_reply.started":"2022-01-02T11:50:28.361464Z","shell.execute_reply":"2022-01-02T11:54:30.056174Z"},"trusted":true},"execution_count":null,"outputs":[]}]}