Commit 538df2a279eb98c22253e368e5c03c964804b2e7

Authored by David Mayerich
1 parent c4980202

added function definitions for new functions. Sebastian and I will finish these definitions today

Showing 3 changed files with 23 additions and 5 deletions   Show diff stats
python/classify.py
... ... @@ -23,4 +23,22 @@ def classcolor2(C):
23 23 color = numpy.asarray(colorsys.hsv_to_rgb(hsv[0], hsv[1], hsv[2])) * 255
24 24 RGB[C[:, :, c], :] = color
25 25  
26   - return RGB
27 26 \ No newline at end of file
  27 + return RGB
  28 +
  29 +#create a function that loads a set of class images as a stack of binary masks
  30 +#input: list of class image names
  31 +#output: X x Y x C stack of binary mask images
  32 +#example: image2class(("class_coll.bmp", "class_epith.bmp"))
  33 +
  34 +#create a set of feature/target pairs for classification
  35 +#input: envi file object, stack of class masks, list of class names
  36 +#output: feature matrix (features x pixels), target matrix (1 x pixels)
  37 +#example: generate_training(("class_coll.bmp", "class_epith.bmp"), (1, 2))
  38 +
  39 +#create a class mask stack from an X x Y x C probability image
  40 +#input: X x Y x C image giving the probability P(c |x,y)
  41 +#output: X x Y x C binary class image
  42 +
  43 +#create an ROC curve calculator
  44 +#input: X x Y x C image giving the probability P(c | x,y)
  45 +#output: ROC curve
28 46 \ No newline at end of file
... ...
python/envi.py
... ... @@ -180,7 +180,6 @@ class envi:
180 180 else:
181 181 self.mask = scipy.misc.imread(maskname, flatten=True).astype(numpy.bool)
182 182  
183   -
184 183 def open(self, filename, headername = ""):
185 184 if headername == "":
186 185 headername = filename + ".hdr"
... ... @@ -256,7 +255,5 @@ class envi:
256 255 bar.update(l+1)
257 256 return M
258 257  
259   -
260   -
261 258 def __del__(self):
262 259 self.file.close()
263 260 \ No newline at end of file
... ...
python/spectral.py
... ... @@ -47,4 +47,7 @@ def unsift2(M, mask):
47 47 else:
48 48 I = numpy.zeros((M.shape[0], mask.shape[0], mask.shape[1]), dtype=M.dtype)
49 49 I[:, i[0], i[1]] = M
50   - return I
51 50 \ No newline at end of file
  51 + return I
  52 +
  53 +#create a function that sifts a color image
  54 +#input: image name, mask
52 55 \ No newline at end of file
... ...