{"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 os\nimport numpy as np\nfrom PIL import Image\nfrom matplotlib import pyplot\nimport io\nimport matplotlib\nimport SimpleITK as sitk\nfrom skimage import io\nfrom tqdm import tqdm\n\npath = ['train_images']\nfor i in path:\n    base_fold = '../input/rsna-2022-cervical-spine-fracture-detection/'+i\n    try:\n        os.mkdir(i)\n    except:\n        pass\n    for folder in tqdm(os.listdir(base_fold)[:1000]):\n        subfolder = os.path.join(base_fold,folder)\n#         print(os.path.join(i,folder))\n        try:\n            os.mkdir(os.path.join(i,folder))\n        except:\n            pass\n        \n        for filename in os.listdir(subfolder):\n#             print(os.path.join(subfolder,filename))\n            \n            itk_img = sitk.ReadImage(os.path.join(subfolder,filename))\n            data_img = sitk.GetArrayFromImage(itk_img)\n            data_img = data_img.transpose(1,2,0)[:,:,0] \n            data_img = Image.fromarray(data_img)\n        \n            stem, suffix = os.path.splitext(filename)\n            stem += '.jpg'\n\n#             matplotlib.image.imsave(os.path.join(os.path.join(i,folder),stem), data_img)\n            matplotlib.image.imsave(os.path.join(os.path.join(i,folder),stem), data_img,cmap=pyplot.cm.bone)\n#             data_img.save(os.path.join(i,folder), Format='jp2')\n\n                 ","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"execution_count":null,"outputs":[]}]}