{"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"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":51753,"databundleVersionId":5692552,"sourceType":"competition"},{"sourceId":6267260,"sourceType":"datasetVersion","datasetId":3467979},{"sourceId":6275345,"sourceType":"datasetVersion","datasetId":3467976}],"dockerImageVersionId":30512,"isInternetEnabled":false,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"! pip install -q --no-index --find-links=file:///kaggle/input/contrails-weights/pip \\\n    segmentation_models_pytorch\n! cp -r /kaggle/input/contrails-src ./src","metadata":{"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"import numpy as np\nimport sys\nimport yaml\nsys.path.append('/kaggle/working/src')\nimport unet5\nimport vit4\nimport unet1024\nimport util\nfrom submit import write_submission","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2023-08-08T22:39:46.711902Z","iopub.execute_input":"2023-08-08T22:39:46.71287Z","iopub.status.idle":"2023-08-08T22:39:53.042675Z","shell.execute_reply.started":"2023-08-08T22:39:46.712829Z","shell.execute_reply":"2023-08-08T22:39:53.041675Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"## Configuration","metadata":{}},{"cell_type":"code","source":"cfg = yaml.safe_load(\"\"\"\ninput:\n  weight: /kaggle/input/contrails-weights\n\nunet5:\n  batch_size: 1\n  models:\n    - name: maxvit_tiny\n      encoder: maxvit_tiny_tf_512.in1k\n      decoder_channels: [256, 128, 64, 32, 16]\n      w: 1.0\n      resize: 512\n      tta: d4prob\n      folds: [0,1,2,3,4]\n\nvit4:\n  batch_size: 2\n  models:\n    - name: vit4_1024\n      encoder: maxvit_tiny_tf_512.in1k\n      decoder_channels: [256, 128, 64, 32, 32]\n      w: 0.6465366913319027\n      #w: 1.0\n      resize: 512\n      tta: d4prob\n      folds: [5, 7]\n\nunet1024:\n  batch_size: 2\n  models:\n    - name: unet1024\n      encoder: maxvit_tiny_tf_512.in1k\n      decoder_channels: [256, 128, 64, 32, 16]\n      w: 0.4531613974630522\n      resize: 1024\n      tta: d4prob\n      folds: [3, 4]\n\"\"\")","metadata":{"execution":{"iopub.status.busy":"2023-08-08T22:39:53.044241Z","iopub.execute_input":"2023-08-08T22:39:53.044606Z","iopub.status.idle":"2023-08-08T22:39:53.05815Z","shell.execute_reply.started":"2023-08-08T22:39:53.044572Z","shell.execute_reply":"2023-08-08T22:39:53.056981Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"## Run","metadata":{}},{"cell_type":"code","source":"data_type = 'test'\n#data_type = 'validation'\ndebug=False\n\n# Unet3 (Ash-color 3-channel image)\n#preds = unet3.run(data_type, cfg, None, debug=debug)\n#preds = unet4.run(data_type, cfg, None, debug=debug)\n#preds = unet5.run(data_type, cfg, None, debug=debug)\n#preds = unet1024.run(data_type, cfg, None, debug=debug)\n\n# Ensemble\npreds = vit4.run(data_type, cfg, None, debug=debug)\nunet1024.run(data_type, cfg, preds, debug=debug)\n\n\nutil.check_preds_finite(preds)\n\n# Write submission.csv\n#th = 0.45\nth = 0.5\nsubmit = write_submission(preds, th, 'submission.csv')","metadata":{"execution":{"iopub.status.busy":"2023-08-08T22:39:53.059873Z","iopub.execute_input":"2023-08-08T22:39:53.060151Z","iopub.status.idle":"2023-08-08T22:40:37.393805Z","shell.execute_reply.started":"2023-08-08T22:39:53.060127Z","shell.execute_reply":"2023-08-08T22:40:37.392424Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"! tail submission.csv\n\nif data_type == 'validation':\n    ! python3 src/score.py submission.csv","metadata":{"execution":{"iopub.status.busy":"2023-08-08T22:40:37.395858Z","iopub.execute_input":"2023-08-08T22:40:37.396265Z","iopub.status.idle":"2023-08-08T22:40:38.388504Z","shell.execute_reply.started":"2023-08-08T22:40:37.396212Z","shell.execute_reply":"2023-08-08T22:40:38.387325Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"","metadata":{},"outputs":[],"execution_count":null}]}