1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
###############################################################################
# pytest_filter2d.py: filter2d
# Author(s): Pieter.Kempeneers@ec.europa.eu
# Copyright (c) 2016-2019 European Union (Joint Research Centre)
# License EUPLv1.2
#
# This file is part of jiplib
###############################################################################
# History
# 2017/11/07 - Created by Pieter Kempeneers (pieter.kempeneers@ec.europa.eu)
# Change log
import argparse
import os
import math
import jiplib as jl
import numpy as np
parser=argparse.ArgumentParser()
parser.add_argument("-input","--input",help="Path of the input raster dataset",dest="input",required=True,type=str)
parser.add_argument("-filter","--filter",help="filter function",dest="filter",required=False,type=str,default='median')
parser.add_argument("-band","--band",help="band to filter (leave empty to filter all bands)",dest="band",required=False,type=int,nargs='+')
parser.add_argument("-size","--size",help="filter size",dest="size",required=False,type=int,default=3)
parser.add_argument("-cut","--cut",help="cut percentage of wavelet coefficients",dest="cut",required=False,type=int,default=0)
parser.add_argument("-nodata","--nodata",help="no data value",dest="nodata",required=False,type=int,nargs='+')
parser.add_argument("-output","--output",help="Path of the classification output raster dataset",dest="output",required=False,type=str)
args = parser.parse_args()
try:
jim=jl.createJim(args.input)
if args.band:
jim=jim.cropBand({'band':args.band})
if 'tap' in args.filter:
taps=np.array([[-1.0,0.0,1.0],[-2.0,0.0,2.0],[-1.0,0.0,1.0]])
fdict={}
fdict.update({'dy': taps.shape[0]})
fdict.update({'dx': taps.shape[1]})
fdict.update({'tap':taps.flatten().tolist()})
fdict.update({'verbose':2})
jim=jim.convert({'otype':'GDT_Float32'})
print(jim.getStats({}))
jim_filtered=jim.filter2d(fdict)
print(jim_filtered.getStats({}))
# jimdx.ngbops.filter2d(tapsdx,nodata=jim_object.properties.getNoDataVals(),abs=True,norm=True)
elif 'dwt' in args.filter:
if args.cut > 0:
jim_filtered=jim.filter2d({'filter':'dwt_cut','otype':'Int16','threshold':args.cut}).pushNoDataValue(0).setThreshold({'min':0,'max':255}).convert({'otype':'Byte'})
else:
jim_filtered=jim.filter2d({'filter':'dwt','otype':'Int16'})
jim_filtered=jim_filtered.filter2d({'filter':'dwti'}).pushNoDataValue(0).setThreshold(0,255).convert({'otype':'Byte'})
else:
if args.nodata:
jim_filtered=jim.filter2d({'filter':args.filter,'dx':args.size,'dy':args.size,'nodata':args.nodata})
else:
jim_filtered=jim.filter2d({'filter':args.filter,'dx':args.size,'dy':args.size})
if args.output:
jim_filtered.write({'filename':args.output,'co':['COMPRESS=LZW','TILED=YES']})
jim_filtered.close()
jim.close()
print("Success: filter2d")
except:
print("Failed: filter2d")
|
