Blame view

python/example.py 774 Bytes
6b2be991   sberisha   added utility fun...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
  import numpy
  import classify
  import matplotlib.pyplot as plt
  from envi import envi
  
  mask_path = '/home/sberisha/data/masks/'
  mask_stack = classify.image2class(mask_path + "class_blood.png", mask_path + "class_coll.png", mask_path +  "class_epith.png",
                           mask_path + "class_lymph.png", mask_path + "class_necrosis.png")
  
  color_image = classify.classcolor2(mask_stack)
  plt.imshow(color_image)
  
  data_path ='/home/sberisha/data/cnn/brc961-nfp8/envi/'
  
  feature_matrix, target_matrix = classify.generate_training(data_path + 'brc961-nfp8-project-br1003', mask_stack)
  
  prob_path = '/home/sberisha/data/'
  prob_envi= envi(prob_path + "cnn-response")
  prob_image = prob_envi.loadall()
  
  class_image = classify.prob2class(prob_image)
  plt.imshow(class_image[4,:,:])