Skip to content
Snippets Groups Projects
run_identify.py 889 B
Newer Older
# Usage Example

from enstools.feature.pipeline import FeaturePipeline
from enstools.feature.identification.african_easterly_waves import AEWIdentification
from enstools.feature.tracking.african_easterly_waves import AEWTracking

from enstools.feature.identification._proto_gen import african_easterly_waves_pb2
pipeline = FeaturePipeline(african_easterly_waves_pb2)

# init AEWIdentification strategy, can take different parameters
i_strat = AEWIdentification()

pipeline.set_identification_strategy(i_strat)
pipeline.set_tracking_strategy(t_strat)
data_path = "/home/christoph/phd/data/aew/cv/*.nc"
pipeline.set_data_path(data_path)

# execute pipeline
pipeline.execute()

# out_netcdf_path = data_path + '_streamers.nc'
out_json_path = data_path[:-4] + 'aew_desc.json'
pipeline.save_result(description_type='json', description_path=out_json_path)