[8271d0]: / notebooks / data.ipynb

Download this file

479 lines (478 with data), 26.4 kB

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "from scipy.misc.pilutil import imresize\n",
    "from glob import glob\n",
    "from utils import get_itk_array"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "nii_path = glob(r'./data_0614/*.nii.gz')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['./data_0614\\\\2020035031_0130_3051_20200130091644_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0130_3051_20200130091644_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0202_3051_20200202100727_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0202_3051_20200202100727_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0205_3051_20200205095731_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0205_3051_20200205095731_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0208_3051_20200208093928_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0208_3051_20200208093928_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0211_3051_20200211112831_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0211_3051_20200211112831_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0214_Lung_Screening_20200214133449_202.nii.gz',\n",
       " './data_0614\\\\2020035031_0214_Lung_Screening_20200214133449_203.nii.gz',\n",
       " './data_0614\\\\2020035031_0224_Chest_Thorax_20200224102911_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0224_Chest_Thorax_20200224102911_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0310_Chest_Thorax_20200310094633_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0310_Chest_Thorax_20200310094633_4.nii.gz',\n",
       " './data_0614\\\\2020035031_0409_Chest_Thorax_20200409095030_2.nii.gz',\n",
       " './data_0614\\\\2020035031_0409_Chest_Thorax_20200409095030_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0130_3050_20200130214643_3.nii.gz',\n",
       " './data_0614\\\\2020035085_0130_3050_20200130214643_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0202_3051_20200202113336_3.nii.gz',\n",
       " './data_0614\\\\2020035085_0202_3051_20200202113336_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0205_3051_20200205113234_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0205_3051_20200205113234_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0208_3051_20200208092321_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0208_3051_20200208092321_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0211_3051_20200211100458_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0211_3051_20200211100458_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0215_3051_20200215105830_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0215_3051_20200215105830_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0219_3051_20200219140720_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0219_3051_20200219140720_4.nii.gz',\n",
       " './data_0614\\\\2020035085_0302_Chest_Thorax_20200302143631_2.nii.gz',\n",
       " './data_0614\\\\2020035085_0302_Chest_Thorax_20200302143631_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0130_3050_20200130233003_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0130_3050_20200130233003_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0202_3052_20200202112749_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0202_3052_20200202112749_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0205_3051_20200205112639_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0205_3051_20200205112639_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0208_3052_20200208090841_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0208_3052_20200208090841_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0211_3051_20200211102410_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0211_3051_20200211102410_4.nii.gz',\n",
       " './data_0614\\\\2020035087_0228_Chest_Thorax_20200228100552_2.nii.gz',\n",
       " './data_0614\\\\2020035087_0228_Chest_Thorax_20200228100552_4.nii.gz',\n",
       " './data_0614\\\\2020035137_0131_3050_20200131140649_2.nii.gz',\n",
       " './data_0614\\\\2020035137_0131_3050_20200131140649_4.nii.gz',\n",
       " './data_0614\\\\2020035137_0203_3052_20200203113456_2.nii.gz',\n",
       " './data_0614\\\\2020035137_0203_3052_20200203113456_4.nii.gz',\n",
       " './data_0614\\\\2020035137_0206_3051_20200206152640_5.nii.gz',\n",
       " './data_0614\\\\2020035137_0206_3051_20200206152640_7.nii.gz',\n",
       " './data_0614\\\\2020035137_0208_3052_20200208091234_2.nii.gz',\n",
       " './data_0614\\\\2020035137_0208_3052_20200208091234_4.nii.gz',\n",
       " './data_0614\\\\2020035137_0211_3051_20200211102118_2.nii.gz',\n",
       " './data_0614\\\\2020035137_0211_3051_20200211102118_4.nii.gz',\n",
       " './data_0614\\\\2020035137_0216_3052_20200216105233_2.nii.gz',\n",
       " './data_0614\\\\2020035137_0216_3052_20200216105233_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0131_3050_20200131153236_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0131_3050_20200131153236_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0203_3052_20200203115050_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0203_3052_20200203115050_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0206_3051_20200206145523_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0206_3051_20200206145523_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0210_3052_20200210093202_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0210_3052_20200210093202_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0214_3052_20200214132354_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0214_3052_20200214132354_4.nii.gz',\n",
       " './data_0614\\\\2020035142_0309_Chest_Thorax_20200309111748_2.nii.gz',\n",
       " './data_0614\\\\2020035142_0309_Chest_Thorax_20200309111748_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0131_3050_20200131205446_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0131_3050_20200131205446_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0203_3051_20200203124259_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0203_3051_20200203124259_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0206_3051_20200206123503_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0206_3051_20200206123503_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0209_3051_20200209102952_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0209_3051_20200209102952_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0212_3051_20200212095002_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0212_3051_20200212095002_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0221_Chest_Thorax_20200221113757_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0221_Chest_Thorax_20200221113757_4.nii.gz',\n",
       " './data_0614\\\\2020035147_0302_Chest_Thorax_20200302105741_2.nii.gz',\n",
       " './data_0614\\\\2020035147_0302_Chest_Thorax_20200302105741_4.nii.gz',\n",
       " './data_0614\\\\2020035153_0131_3050_20200131180142_2.nii.gz',\n",
       " './data_0614\\\\2020035153_0131_3050_20200131180142_4.nii.gz',\n",
       " './data_0614\\\\2020035153_0203_3051_20200203122757_2.nii.gz',\n",
       " './data_0614\\\\2020035153_0203_3051_20200203122757_4.nii.gz',\n",
       " './data_0614\\\\2020035153_0206_3051_20200206150738_3.nii.gz',\n",
       " './data_0614\\\\2020035153_0206_3051_20200206150738_5.nii.gz',\n",
       " './data_0614\\\\2020035153_0209_3051_20200209093133_2.nii.gz',\n",
       " './data_0614\\\\2020035153_0209_3051_20200209093133_4.nii.gz',\n",
       " './data_0614\\\\2020035153_0214_3052_20200214131154_2.nii.gz',\n",
       " './data_0614\\\\2020035153_0214_3052_20200214131154_4.nii.gz',\n",
       " './data_0614\\\\2020035153_0224_Chest_Thorax_20200224132213_2.nii.gz',\n",
       " './data_0614\\\\2020035153_0224_Chest_Thorax_20200224132213_4.nii.gz',\n",
       " './data_0614\\\\2020035201_0201_3050_20200201143734_2.nii.gz',\n",
       " './data_0614\\\\2020035201_0201_3050_20200201143734_4.nii.gz',\n",
       " './data_0614\\\\2020035201_0204_3051_20200204100643_2.nii.gz',\n",
       " './data_0614\\\\2020035201_0204_3051_20200204100643_4.nii.gz',\n",
       " './data_0614\\\\2020035201_0207_3052_20200207120329_2.nii.gz',\n",
       " './data_0614\\\\2020035201_0207_3052_20200207120329_4.nii.gz',\n",
       " './data_0614\\\\2020035201_0210_3052_20200210101216_2.nii.gz',\n",
       " './data_0614\\\\2020035201_0210_3052_20200210101216_4.nii.gz',\n",
       " './data_0614\\\\2020035201_0310_Chest_Thorax_20200310152721_2.nii.gz',\n",
       " './data_0614\\\\2020035201_0310_Chest_Thorax_20200310152721_4.nii.gz',\n",
       " './data_0614\\\\2020035205_0201_3050_20200201132923_2.nii.gz',\n",
       " './data_0614\\\\2020035205_0201_3050_20200201132923_4.nii.gz',\n",
       " './data_0614\\\\2020035205_0204_3050_20200204145517_2.nii.gz',\n",
       " './data_0614\\\\2020035205_0204_3050_20200204145517_4.nii.gz',\n",
       " './data_0614\\\\2020035205_0208_3052_20200208175643_2.nii.gz',\n",
       " './data_0614\\\\2020035205_0208_3052_20200208175643_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0201_3050_20200201185149_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0201_3050_20200201185149_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0204_3051_20200204094049_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0204_3051_20200204094049_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0207_3051_20200207121831_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0207_3051_20200207121831_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0210_3051_20200210104039_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0210_3051_20200210104039_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0302_Chest_Thorax_20200302100829_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0302_Chest_Thorax_20200302100829_4.nii.gz',\n",
       " './data_0614\\\\2020035210_0324_Chest_Thorax_20200324091948_2.nii.gz',\n",
       " './data_0614\\\\2020035210_0324_Chest_Thorax_20200324091948_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0202_3050_20200202113720_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0202_3050_20200202113720_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0206_3051_20200206153916_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0206_3051_20200206153916_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0209_3052_20200209111756_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0209_3052_20200209111756_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0212_3052_20200212091422_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0212_3052_20200212091422_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0221_Chest_Thorax_20200221091847_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0221_Chest_Thorax_20200221091847_4.nii.gz',\n",
       " './data_0614\\\\2020035224_0302_Chest_Thorax_20200302095843_2.nii.gz',\n",
       " './data_0614\\\\2020035224_0302_Chest_Thorax_20200302095843_4.nii.gz',\n",
       " './data_0614\\\\2020035227_0202_3050_20200202114502_2.nii.gz',\n",
       " './data_0614\\\\2020035227_0202_3050_20200202114502_4.nii.gz',\n",
       " './data_0614\\\\2020035227_0205_3051_20200205122053_2.nii.gz',\n",
       " './data_0614\\\\2020035227_0205_3051_20200205122053_4.nii.gz',\n",
       " './data_0614\\\\2020035227_0210_3051_20200210154805_2.nii.gz',\n",
       " './data_0614\\\\2020035227_0210_3051_20200210154805_4.nii.gz',\n",
       " './data_0614\\\\2020035227_0213_3051_20200213110520_2.nii.gz',\n",
       " './data_0614\\\\2020035227_0213_3051_20200213110520_4.nii.gz',\n",
       " './data_0614\\\\2020035227_0217_3051_20200217131402_3.nii.gz',\n",
       " './data_0614\\\\2020035227_0217_3051_20200217131402_5.nii.gz',\n",
       " './data_0614\\\\2020035227_0218_3051_20200218101050_2.nii.gz',\n",
       " './data_0614\\\\2020035227_0218_3051_20200218101050_4.nii.gz',\n",
       " './data_0614\\\\2020035234_0202_3050_20200202181926_2.nii.gz',\n",
       " './data_0614\\\\2020035234_0202_3050_20200202181926_4.nii.gz',\n",
       " './data_0614\\\\2020035234_0205_3051_20200205133130_2.nii.gz',\n",
       " './data_0614\\\\2020035234_0205_3051_20200205133130_4.nii.gz',\n",
       " './data_0614\\\\2020035234_0326_2917_20200326162248_11.nii.gz',\n",
       " './data_0614\\\\2020035234_0326_2917_20200326162248_9.nii.gz',\n",
       " './data_0614\\\\2020035234_0401_3046_20200401084903_3.nii.gz',\n",
       " './data_0614\\\\2020035234_0401_3046_20200401084903_6.nii.gz',\n",
       " './data_0614\\\\2020035234_0413_3051_20200413082141_2.nii.gz',\n",
       " './data_0614\\\\2020035234_0413_3051_20200413082141_4.nii.gz',\n",
       " './data_0614\\\\2020035234_0429_Chest_Thorax_20200429142024_2.nii.gz',\n",
       " './data_0614\\\\2020035234_0429_Chest_Thorax_20200429142024_4.nii.gz',\n",
       " './data_0614\\\\2020035238_0202_3050_20200202232954_3.nii.gz',\n",
       " './data_0614\\\\2020035238_0202_3050_20200202232954_5.nii.gz',\n",
       " './data_0614\\\\2020035238_0206_3051_20200206102224_2.nii.gz',\n",
       " './data_0614\\\\2020035238_0206_3051_20200206102224_4.nii.gz',\n",
       " './data_0614\\\\2020035238_0209_3052_20200209142019_2.nii.gz',\n",
       " './data_0614\\\\2020035238_0209_3052_20200209142019_4.nii.gz',\n",
       " './data_0614\\\\2020035240_0203_3050_20200203022236_2.nii.gz',\n",
       " './data_0614\\\\2020035240_0203_3050_20200203022236_4.nii.gz',\n",
       " './data_0614\\\\2020035240_0206_3051_20200206112432_2.nii.gz',\n",
       " './data_0614\\\\2020035240_0206_3051_20200206112432_4.nii.gz',\n",
       " './data_0614\\\\2020035240_0208_3052_20200208151949_5.nii.gz',\n",
       " './data_0614\\\\2020035240_0208_3052_20200208151949_7.nii.gz',\n",
       " './data_0614\\\\2020035362_0204_3050_20200204150454_2.nii.gz',\n",
       " './data_0614\\\\2020035362_0204_3050_20200204150454_4.nii.gz',\n",
       " './data_0614\\\\2020035362_0207_3051_20200207152308_2.nii.gz',\n",
       " './data_0614\\\\2020035362_0207_3051_20200207152308_4.nii.gz',\n",
       " './data_0614\\\\2020035362_0210_Lung_Screening_20200210142745_201.nii.gz',\n",
       " './data_0614\\\\2020035362_0210_Lung_Screening_20200210142745_203.nii.gz',\n",
       " './data_0614\\\\2020035362_0213_Chest_Helical_20200213091206_201.nii.gz',\n",
       " './data_0614\\\\2020035362_0213_Chest_Helical_20200213091206_203.nii.gz',\n",
       " './data_0614\\\\2020035362_0228_Chest_Thorax_20200228104648_2.nii.gz',\n",
       " './data_0614\\\\2020035362_0228_Chest_Thorax_20200228104648_4.nii.gz',\n",
       " './data_0614\\\\2020035365_0204_3050_20200204184413_2.nii.gz',\n",
       " './data_0614\\\\2020035365_0204_3050_20200204184413_4.nii.gz',\n",
       " './data_0614\\\\2020035365_0207_3052_20200207110908_2.nii.gz',\n",
       " './data_0614\\\\2020035365_0207_3052_20200207110908_4.nii.gz',\n",
       " './data_0614\\\\2020035365_0209_3052_20200209161622_2.nii.gz',\n",
       " './data_0614\\\\2020035365_0209_3052_20200209161622_4.nii.gz',\n",
       " './data_0614\\\\2020035365_0212_Lung_Screening_20200212101052_201.nii.gz',\n",
       " './data_0614\\\\2020035365_0212_Lung_Screening_20200212101052_203.nii.gz',\n",
       " './data_0614\\\\2020035365_0215_Lung_Screening_20200215100533_201.nii.gz',\n",
       " './data_0614\\\\2020035365_0215_Lung_Screening_20200215100533_203.nii.gz',\n",
       " './data_0614\\\\2020035365_0218_Lung_Screening_20200218091424_201.nii.gz',\n",
       " './data_0614\\\\2020035365_0218_Lung_Screening_20200218091424_203.nii.gz',\n",
       " './data_0614\\\\2020035365_0304_Chest_Thorax_20200304110557_2.nii.gz',\n",
       " './data_0614\\\\2020035365_0304_Chest_Thorax_20200304110557_4.nii.gz',\n",
       " './data_0614\\\\2020035365_0320_Chest_Thorax_20200320101254_2.nii.gz',\n",
       " './data_0614\\\\2020035365_0320_Chest_Thorax_20200320101254_4.nii.gz',\n",
       " './data_0614\\\\2020035395_0205_3050_20200205122634_2.nii.gz',\n",
       " './data_0614\\\\2020035395_0205_3050_20200205122634_4.nii.gz',\n",
       " './data_0614\\\\2020035395_0208_3051_20200208213723_2.nii.gz',\n",
       " './data_0614\\\\2020035395_0208_3051_20200208213723_4.nii.gz',\n",
       " './data_0614\\\\2020035395_0211_Lung_Screening_20200211142605_201.nii.gz',\n",
       " './data_0614\\\\2020035395_0211_Lung_Screening_20200211142605_203.nii.gz',\n",
       " './data_0614\\\\2020035395_0214_Lung_Screening_20200214141904_201.nii.gz',\n",
       " './data_0614\\\\2020035395_0214_Lung_Screening_20200214141904_203.nii.gz',\n",
       " './data_0614\\\\2020035395_0216_Lung_Screening_20200216103047_201.nii.gz',\n",
       " './data_0614\\\\2020035395_0216_Lung_Screening_20200216103047_203.nii.gz',\n",
       " './data_0614\\\\2020035395_0303_Chest_Thorax_20200303150658_2.nii.gz',\n",
       " './data_0614\\\\2020035395_0303_Chest_Thorax_20200303150658_4.nii.gz',\n",
       " './data_0614\\\\2020035396_0205_3050_20200205124403_2.nii.gz',\n",
       " './data_0614\\\\2020035396_0205_3050_20200205124403_4.nii.gz',\n",
       " './data_0614\\\\2020035396_0208_3052_20200208203552_2.nii.gz',\n",
       " './data_0614\\\\2020035396_0208_3052_20200208203552_4.nii.gz',\n",
       " './data_0614\\\\2020035396_0211_Lung_Screening_20200211143300_201.nii.gz',\n",
       " './data_0614\\\\2020035396_0211_Lung_Screening_20200211143300_203.nii.gz',\n",
       " './data_0614\\\\2020035396_0213_Lung_Screening_20200213135411_201.nii.gz',\n",
       " './data_0614\\\\2020035396_0213_Lung_Screening_20200213135411_203.nii.gz',\n",
       " './data_0614\\\\2020035396_0216_Lung_Screening_20200216102811_201.nii.gz',\n",
       " './data_0614\\\\2020035396_0216_Lung_Screening_20200216102811_203.nii.gz',\n",
       " './data_0614\\\\2020035396_0303_Chest_Thorax_20200303150020_2.nii.gz',\n",
       " './data_0614\\\\2020035396_0303_Chest_Thorax_20200303150020_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0214_3050_20200214220726_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0214_3050_20200214220726_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0217_3052_20200217104646_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0217_3052_20200217104646_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0223_3051_20200223103122_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0223_3051_20200223103122_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0227_3051_20200227095603_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0227_3051_20200227095603_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0304_3051_20200304150058_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0304_3051_20200304150058_4.nii.gz',\n",
       " './data_0614\\\\2020035751_0319_Chest_Thorax_20200319105532_2.nii.gz',\n",
       " './data_0614\\\\2020035751_0319_Chest_Thorax_20200319105532_4.nii.gz',\n",
       " './data_0614\\\\2020035775_0215_3050_20200215221834_2.nii.gz',\n",
       " './data_0614\\\\2020035775_0215_3050_20200215221834_4.nii.gz',\n",
       " './data_0614\\\\2020035775_0218_Lung_Screening_20200218084514_201.nii.gz',\n",
       " './data_0614\\\\2020035775_0218_Lung_Screening_20200218084514_203.nii.gz',\n",
       " './data_0614\\\\2020035775_0221_3051_20200221131815_3.nii.gz',\n",
       " './data_0614\\\\2020035775_0221_3051_20200221131815_5.nii.gz',\n",
       " './data_0614\\\\2020035775_0224_3051_20200224092851_2.nii.gz',\n",
       " './data_0614\\\\2020035775_0224_3051_20200224092851_4.nii.gz',\n",
       " './data_0614\\\\2020035775_0302_3051_20200302133455_2.nii.gz',\n",
       " './data_0614\\\\2020035775_0302_3051_20200302133455_4.nii.gz',\n",
       " './data_0614\\\\2020035775_0313_Chest_Thorax_20200313093733_2.nii.gz',\n",
       " './data_0614\\\\2020035775_0313_Chest_Thorax_20200313093733_4.nii.gz',\n",
       " './data_0614\\\\2020035775_0327_Chest_Thorax_20200327091704_2.nii.gz',\n",
       " './data_0614\\\\2020035775_0327_Chest_Thorax_20200327091704_4.nii.gz',\n",
       " './data_0614\\\\2020036063_0222_3050_20200222144948_2.nii.gz',\n",
       " './data_0614\\\\2020036063_0222_3050_20200222144948_4.nii.gz',\n",
       " './data_0614\\\\2020036063_0225_3052_20200225103239_2.nii.gz',\n",
       " './data_0614\\\\2020036063_0225_3052_20200225103239_4.nii.gz',\n",
       " './data_0614\\\\2020036063_0229_3052_20200229105243_2.nii.gz',\n",
       " './data_0614\\\\2020036063_0229_3052_20200229105243_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0226_3050_20200226235401_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0226_3050_20200226235401_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0229_3051_20200229104921_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0229_3051_20200229104921_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0304_3051_20200304145339_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0304_3051_20200304145339_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0312_Chest_Thorax_20200312104101_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0312_Chest_Thorax_20200312104101_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0326_Chest_Thorax_20200326095432_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0326_Chest_Thorax_20200326095432_4.nii.gz']"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "nii_path"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 104,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_path = nii_path[254:254+10]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 105,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['./data_0614\\\\2020036291_0226_3050_20200226235401_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0226_3050_20200226235401_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0229_3051_20200229104921_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0229_3051_20200229104921_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0304_3051_20200304145339_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0304_3051_20200304145339_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0312_Chest_Thorax_20200312104101_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0312_Chest_Thorax_20200312104101_4.nii.gz',\n",
       " './data_0614\\\\2020036291_0326_Chest_Thorax_20200326095432_2.nii.gz',\n",
       " './data_0614\\\\2020036291_0326_Chest_Thorax_20200326095432_4.nii.gz']"
      ]
     },
     "execution_count": 105,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_path"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 106,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "matrix shape:\t (77, 512, 512)\n",
      "matrix shape:\t (474, 512, 512)\n",
      "matrix shape:\t (77, 512, 512)\n",
      "matrix shape:\t (560, 512, 512)\n",
      "matrix shape:\t (77, 512, 512)\n",
      "matrix shape:\t (560, 512, 512)\n",
      "matrix shape:\t (71, 512, 512)\n",
      "matrix shape:\t (438, 512, 512)\n",
      "matrix shape:\t (73, 512, 512)\n",
      "matrix shape:\t (451, 512, 512)\n"
     ]
    }
   ],
   "source": [
    "tag = 1\n",
    "meta_data = []\n",
    "for name in test_path:\n",
    "    nii, spacing = get_itk_array(name)\n",
    "    patientID = os.path.basename(name).split('.')[0].split('_')[0]\n",
    "    print('matrix shape:\\t', nii.shape)\n",
    "    nii[nii < -1000] = -1000\n",
    "    nii[nii > 512] = 512\n",
    "    nii = nii - np.min(nii)\n",
    "    nii = nii * 1.0 / np.max(nii)\n",
    "    if tag == 1:\n",
    "        total = nii\n",
    "        tag = 0\n",
    "    else:\n",
    "        total = np.concatenate((total, nii), axis=0)\n",
    "    meta_data.append({'patientID': patientID, 'filename': name, 'slice': nii.shape[0], 'spacing': spacing, 'shape': nii.shape})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 107,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\hasee007\\AppData\\Roaming\\Python\\Python37\\site-packages\\ipykernel_launcher.py:4: DeprecationWarning: `imresize` is deprecated!\n",
      "`imresize` is deprecated in SciPy 1.0.0, and will be removed in 1.2.0.\n",
      "Use ``skimage.transform.resize`` instead.\n",
      "  after removing the cwd from sys.path.\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "sacle of nii is changed after imresize!\n",
      " Now, normlise:\n"
     ]
    }
   ],
   "source": [
    "slices = total.shape[0]\n",
    "total_all = np.zeros((slices, 256, 256))  # 初始化\n",
    "for i in range(slices):\n",
    "    total_all[i, :, :] = imresize(total[i, :, :], (256, 256))  # 缩小尺寸\n",
    "if np.max(total_all) > 1:\n",
    "    print('sacle of nii is changed after imresize!\\n Now, normlise:')\n",
    "    # normalise to 0-1\n",
    "    total_all = total_all - np.min(total_all)\n",
    "    total_all = total_all * 1.0 / np.max(total_all)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 108,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "matrix shape:\t (2858, 256, 256)\n"
     ]
    }
   ],
   "source": [
    "npy_name = name.split('\\\\')[-1].split('.')[0].split('_')[0]\n",
    "np.save('{}.npy'.format(npy_name), total_all)\n",
    "df = pd.DataFrame(meta_data)\n",
    "df.to_csv('{}.csv'.format(npy_name))\n",
    "print('matrix shape:\\t', total_all.shape)  # (301, 512, 512)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}