{"metadata": {"_change_revision": 0, "_is_fork": false, "language_info": {"name": "python", "nbconvert_exporter": "python", "file_extension": ".py", "codemirror_mode": {"name": "ipython", "version": 3}, "mimetype": "text/x-python", "version": "3.6.1", "pygments_lexer": "ipython3"}, "kernelspec": {"name": "python3", "display_name": "Python 3", "language": "python"}}, "nbformat": 4, "cells": [{"cell_type": "code", "execution_count": null, "metadata": {"trusted": false, "_cell_guid": "0f245602-1b68-4fb9-f25b-959673f80fa6", "_uuid": "5ffff5941f98de7e654c8fc1683a8e26ffe5000f", "_execution_state": "idle"}, "outputs": [], "source": "import os\nimport numpy as np\nimport tifffile as tiff\nimport cv2\nimport matplotlib.pyplot as plt\n\n\ndef stretch_8bit(bands, lower_percent=2, higher_percent=98):\n    out = np.zeros_like(bands)\n    for i in range(3):\n        a = 0 \n        b = 255 \n        c = np.percentile(bands[:,:,i], lower_percent)\n        print (c)\n        d = np.percentile(bands[:,:,i], higher_percent) \n        print (d)\n        t = a + (bands[:,:,i] - c) * (b - a) / (d - c)    \n        t[t<a] = a\n        t[t>b] = b\n        out[:,:,i] =t\n    return out.astype(np.uint8)    \n    \ndef M(image_id):\n    filename = os.path.join('..', 'input', 'sixteen_band', '{}_M.tif'.format(image_id))\n    img = tiff.imread(filename)\n    print (img.shape)\n    img = np.rollaxis(img, 0, 3)\n    print (img.shape)\n    return img\n\nimage_id = '6120_2_2'\nm = M(image_id)\nimg = np.zeros((837,851,3))\nimg[:,:,0] = m[:,:,4] #red\nimg[:,:,1] = m[:,:,2] #green\nimg[:,:,2] = m[:,:,1] #blue\n#plt.imshow(img)\nplt.imshow(stretch_8bit(img))\n\n"}], "nbformat_minor": 0}