{"cells":[{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nimport pydicom\nimport matplotlib.pyplot as plt\nfrom matplotlib.patches import Polygon","execution_count":null,"outputs":[]},{"metadata":{"_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","trusted":true},"cell_type":"code","source":"def draw_boxes(image, boxes):\n    fig, ax = plt.subplots(1, figsize=(7,7))\n    \n    colors = {i: np.random.rand(3,) for i in range(len(boxes))}\n    for i, box in enumerate(boxes):\n        y, x, h, w = box[0], box[1], box[2], box[3]\n        ymin, xmin, ymax, xmax = x, y, x+w, y+h\n        p = Polygon(((xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)),\n            fc=(colors[i][0],colors[i][1],colors[i][2],0.35), \n            ec=(colors[i][0],colors[i][1],colors[i][2],0.95), lw=3)\n        ax.add_patch(p)\n    ax.imshow(image, cmap=plt.cm.gist_gray)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true,"_uuid":"0e7bf658a4f96ca2eb60313e982b00fce800ef98"},"cell_type":"code","source":"train_info_path = '../input/stage_1_train_labels.csv'\ndata = pd.read_csv(train_info_path)\n\n# get labels\nidx = '00c0b293-48e7-4e16-ac76-9269ba535a62'\nboxes = data[data['patientId'] == idx][['x', 'y', 'width', 'height']].values.tolist()\n\n# get image\ndcm_data = pydicom.read_file('../input/stage_1_train_images/'+idx+'.dcm')\nimage = dcm_data.pixel_array\n\n# draw image with labels\ndraw_boxes(image, boxes)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true,"_uuid":"6f3536312debe142b814f5dda9f24cae34a5538b"},"cell_type":"code","source":"","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"name":"python","version":"3.6.6","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"}},"nbformat":4,"nbformat_minor":1}