Commit eccf10ffba69bc9d89f04fdbbb4237dd7a2d6ae3
1 parent
254e3bc8
added comments for the envi classification function
Showing
1 changed file
with
4 additions
and
0 deletions
Show diff stats
python/classify.py
@@ -159,6 +159,10 @@ def random_mask(M, n): | @@ -159,6 +159,10 @@ def random_mask(M, n): | ||
159 | new_mask[numpy.unravel_index(new_idx[0:n], new_mask.shape)] = True | 159 | new_mask[numpy.unravel_index(new_idx[0:n], new_mask.shape)] = True |
160 | return new_mask | 160 | return new_mask |
161 | 161 | ||
162 | +#perform classification of an ENVI image using batch processing | ||
163 | +# input: E is the ENVI object (file is assumed to be loaded) | ||
164 | +# C is a classifier - anything in sklearn should work | ||
165 | +# batch is the batch size | ||
162 | def envi_batch_predict(E, C, batch=10000): | 166 | def envi_batch_predict(E, C, batch=10000): |
163 | 167 | ||
164 | Fv = E.loadbatch(batch) | 168 | Fv = E.loadbatch(batch) |