{"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 pydicom\nimport numpy as np\nimport cv2\nimport os\nfrom joblib import Parallel, delayed\nfrom tqdm.notebook import tqdm\nfrom pathlib import Path\n\nRESIZE_TO = (1024, 1024)\n\n!rm -rf test_images_processed_cv2_{RESIZE_TO[0]}\n!mkdir test_images_processed_cv2_{RESIZE_TO[0]}\n\n# https://www.kaggle.com/code/tanlikesmath/brain-tumor-radiogenomic-classification-eda/notebook\ndef dicom_file_to_ary(path):\n    dicom = pydicom.dcmread(path)\n    data = dicom.pixel_array\n       \n    data = (data - data.min()) / (data.max() - data.min())\n    \n    if dicom.PhotometricInterpretation == \"MONOCHROME1\":\n        data = 1 - data\n        \n    data = cv2.resize(data, RESIZE_TO)\n    data = (data * 255).astype(np.uint8)\n    return data\n\ndirectories = list(Path('/kaggle/input/rsna-breast-cancer-detection/test_images').iterdir())\n\ndef process_directory(directory_path):\n    parent_directory = str(directory_path).split('/')[-1]\n    !mkdir -p test_images_processed_cv2_{RESIZE_TO[0]}/{parent_directory}\n    for image_path in directory_path.iterdir():\n        processed_ary = dicom_file_to_ary(image_path)\n        \n        cv2.imwrite(\n            f'test_images_processed_cv2_{RESIZE_TO[0]}/{parent_directory}/{image_path.stem}.png',\n            processed_ary\n        )\n        \nimport multiprocessing as mp\n\nwith mp.Pool(64) as p:\n    p.map(process_directory, directories)","metadata":{"execution":{"iopub.status.busy":"2023-02-06T00:53:14.934955Z","iopub.execute_input":"2023-02-06T00:53:14.935419Z","iopub.status.idle":"2023-02-06T00:53:19.343526Z","shell.execute_reply.started":"2023-02-06T00:53:14.935338Z","shell.execute_reply":"2023-02-06T00:53:19.342362Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"","metadata":{},"execution_count":null,"outputs":[]}]}