In [1]:

    

%load_ext autoreload
%autoreload 2


    

In [5]:

    

import os
#import warnings
#warnings.simplefilter('ignore')
from operator import attrgetter
from platform import python_version_tuple

if python_version_tuple()[0] == 3:
    xrange = range

import scipy as sp
import scipy.ndimage
import numpy as np
import pandas as pd
import skimage
import skimage.measure
from PIL import Image
import matplotlib.pyplot as plt
%matplotlib inline
import cv2
import h5py
from tqdm import tqdm_notebook
from IPython.display import display
from extract_data import *


    

In [6]:

    

h5f = h5py.File(os.path.join(os.path.dirname(os.getcwd()), 'data', 'hdf5_datasets', 'all_data.hdf5'), 'r+')


    

In [3]:

    

X, Y, filenames = extract_DRIONS_DB(os.path.join(os.path.dirname(os.getcwd()), 'data', 'DRIONS-DB'), expert=1)
len(X), X[0].shape, len(Y), Y[0].shape, len(filenames)


    

    
Out[3]:





(110, (560, 560, 3), 110, (560, 560, 1), 110)

In [41]:

    

for i in xrange(10):
    cur = X[i].copy()
    cur[np.where(Y[i])] /= 4
    plt.imshow(X[i])
    plt.show()
    plt.imshow(cur)
    plt.show()


    

In [10]:

    

X, Y, filenames, is_ill = extract_RIM_ONE_v2(os.path.join(os.path.dirname(os.getcwd()), 'data', 'RIM-ONE v2'))
len(X), X[0].shape, len(Y), Y[0].shape, len(filenames), len(is_ill)


    

    
Out[10]:





(455, (394, 394, 3), 455, (394, 394, 1), 455, 455)

In [12]:

    

for i in xrange(10):
    cur = X[i].copy()
    cur[np.where(Y[i])] /= 4
    plt.imshow(X[i])
    plt.show()
    plt.imshow(cur)
    plt.show()


    

In [7]:

    

X, disc, cup, filenames, is_ill = extract_RIM_ONE_v3(os.path.join(os.path.dirname(os.getcwd()), 'data', 'RIM-ONE v3'))
len(X), X[0].shape, len(disc), disc[0].shape, len(cup), cup[0].shape, \
    len(filenames), len(is_ill)


    

    
Out[7]:





(159, (1424, 1424, 3), 159, (1424, 1424, 1), 159, (1424, 1424, 1), 159, 159)

In [8]:

    

for i in xrange(10):
    print(filenames[i])
    cur = X[i].copy()
    cur[np.where(disc[i])] /= 4
    plt.imshow(X[i])
    plt.show()
    plt.imshow(cur)
    plt.show()


    

    




N-1-L






    













    













    




N-10-R






    













    













    




N-11-L






    













    













    




N-12-R






    













    













    




N-13-L






    













    













    




N-14-R






    













    













    




N-16-R






    













    













    




N-17-L






    













    













    




N-18-R






    













    













    




N-2-R