{"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\nimport pandas as pd\nimport os\nimport cv2\nimport numpy as np\nimport glob\nfrom tqdm import tqdm","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2021-07-27T07:56:47.878732Z","iopub.execute_input":"2021-07-27T07:56:47.879167Z","iopub.status.idle":"2021-07-27T07:56:47.884717Z","shell.execute_reply.started":"2021-07-27T07:56:47.879124Z","shell.execute_reply":"2021-07-27T07:56:47.883349Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"for dir_ in sorted(glob.glob('../input/rsna-miccai-png/*/*')):\n    train, id_ = (dir_.split('/')[-2:])\n    try: os.makedirs(f'dataset/{train}/{id_}')\n    except: pass","metadata":{"execution":{"iopub.status.busy":"2021-07-27T07:56:48.104168Z","iopub.execute_input":"2021-07-27T07:56:48.104916Z","iopub.status.idle":"2021-07-27T07:56:48.138007Z","shell.execute_reply.started":"2021-07-27T07:56:48.104853Z","shell.execute_reply":"2021-07-27T07:56:48.136559Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"font = cv2.FONT_HERSHEY_SIMPLEX\n\nfor dir_ in tqdm(glob.glob('../input/rsna-miccai-png/*/*/*')):\n    train, id_, name_ = dir_.split('/')[-3:]\n    save_vid = f'dataset/{train}/{id_}/{name_}.avi'\n\n    img_array = []\n    d = {int(image.split('/')[-1].split('.')[0].split('-')[-1]):image for image in sorted(glob.glob(f'{dir_}/*.png'))}\n    for num, filename in sorted(d.items()):\n        img = cv2.imread(filename)\n        cv2.putText(img,f'{num}',(10,500), font, 1,(255,255,255),2)\n        height, width, layers = img.shape\n        size = (width,height)\n        img_array.append(img)\n\n    out = cv2.VideoWriter(save_vid, cv2.VideoWriter_fourcc(*'DIVX'), 5, size)\n    for i in range(len(img_array)):\n        out.write(img_array[i])\n    out.release()","metadata":{"execution":{"iopub.status.busy":"2021-07-27T07:56:48.75569Z","iopub.execute_input":"2021-07-27T07:56:48.756094Z","iopub.status.idle":"2021-07-27T08:42:11.935643Z","shell.execute_reply.started":"2021-07-27T07:56:48.756042Z","shell.execute_reply":"2021-07-27T08:42:11.934186Z"},"trusted":true},"execution_count":null,"outputs":[]}]}