geowatch.tasks.cold.predict module¶

Main prediction script for cold

SeeAlso:

../../cli/queue_cli/prepare_teamfeats.py

predict.py *

prepare_kwcoco.py

tile_processing_kwcoco.py

export_cold_result_kwcoco.py

assemble_cold_result_kwcoco.py

CommandLine

###################################################################################
### FULL REGION TEST: COLD FEATURES WITH HIGH TEMPORAL RESOLUTION (HTR) + L8/S2 ###
###################################################################################

DATA_DVC_DPATH=$(geowatch_dvc --tags=phase3_data --hardware="auto")
EXPT_DVC_DPATH=$(geowatch_dvc --tags=phase3_expt --hardware="auto")
python -m geowatch.tasks.cold.predict \
    --coco_fpath="$DATA_DVC_DPATH/Aligned-Drop8-ARA/KR_R001/imgonly-KR_R001.kwcoco.zip" \
    --out_dpath="$DATA_DVC_DPATH/Aligned-Drop8-ARA/_pycold_Drop8" \
    --mod_coco_fpath="$DATA_DVC_DPATH/Aligned-Drop8-ARA/KR_R001/imgonly_KR_R001_cold-biyearly.kwcoco.zip" \
    --sensors='L8,S2' \
    --coefs=cv,rmse,a0,a1,b1,c1 \
    --prob=0.99 \
    --conse=8 \
    --coefs_bands=0,1,2,3,4,5 \
    --combine=False \
    --resolution='10GSD' \
    --cold_time_span='6months' \
    --workermode='process' \
    --workers=8

######################################################################
### FULL REGION TEST: TRANSFER COLD FEATURE FROM RAW TO COMBINED INPUT
######################################################################

DATA_DVC_DPATH=$(geowatch_dvc --tags=phase3_data --hardware="auto")
EXPT_DVC_DPATH=$(geowatch_dvc --tags=phase3_expt --hardware="auto")
python -m geowatch.tasks.cold.transfer_features \
    --coco_fpath="$DATA_DVC_DPATH/Aligned-Drop8-ARA/KR_R001/imgonly_KR_R001_cold-biyearly.kwcoco.zip" \
    --combine_fpath="$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imgonly-KR_R001.kwcoco.zip" \
    --new_coco_fpath="$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imganns-KR_R001-cold-biyearly.kwcoco.zip"

kwcoco stats "$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imganns-KR_R001-cold-biyearly.kwcoco.zip"
geowatch stats "$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imganns-KR_R001-cold-biyearly.kwcoco.zip"
kwcoco validate "$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imganns-KR_R001-cold-biyearly.kwcoco.zip"

DATA_DVC_DPATH=$(geowatch_dvc --tags=phase3_data --hardware="auto")
geowatch visualize \
    "$DATA_DVC_DPATH/Drop8-Median10GSD-V1/imganns-KR_R001-cold-biyearly.kwcoco.zip" \
    --channels="L8:(red|green|blue,red_COLD_a1|green_COLD_a1|blue_COLD_a1,red_COLD_cv|green_COLD_cv|blue_COLD_cv,red_COLD_rmse|green_COLD_rmse|blue_COLD_rmse)" \
    --exclude_sensors=WV,PD,S2 \
    --smart=True

########################
### MULTIPLE REGION TEST
########################
DVC_DATA_DPATH=$(geowatch_dvc --tags='phase2_data' --hardware=auto)

"$BUNDLE_DPATH"/imganns-*BR_[RC]*.kwcoco.zip \
"$BUNDLE_DPATH"/imganns-*KR_[RC]*.kwcoco.zip \
"$BUNDLE_DPATH"/imganns-*NZ_[RC]*.kwcoco.zip \
"$BUNDLE_DPATH"/imganns-*US_[RC]*.kwcoco.zip \

echo "$DVC_DATA_DPATH"
BUNDLE_DPATH=$DVC_DATA_DPATH/Drop6
python -m geowatch.cli.queue_cli.prepare_teamfeats \
    --base_fpath \
        "$BUNDLE_DPATH"/imganns-*AE_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*BH_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*CH_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*LT_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*NZ_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*PE_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*QA_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*SA_[RC]*.kwcoco.zip \
        "$BUNDLE_DPATH"/imganns-*US_C*.kwcoco.zip \
    --with_cold=1 \
    --with_landcover=0 \
    --with_materials=0 \
    --with_invariants=0 \
    --with_depth=0 \
    --skip_existing=1 \
    --cold_workers=8 \
    --cold_workermode=thread \
    --tmux_workers=2 \
    --backend=tmux --run=0

class geowatch.tasks.cold.predict.ColdPredictConfig(*args, **kwargs)[source]¶

Bases: DataConfig

The docstring will be the description in the CLI help

Valid options: []

Parameters:

*args – positional arguments for this data config
**kwargs – keyword arguments for this data config

default = {'adj_cloud': <Value(False)>, 'cm_interval': <Value(60)>, 'coco_fpath': <Value(None)>, 'coefs': <Value(None)>, 'coefs_bands': <Value(None)>, 'cold_time_span': <Value('1year')>, 'combine': <Value(False)>, 'combined_coco_fpath': <Value(None)>, 'conse': <Value(6)>, 'exclude_first': <Value(True)>, 'method': <Value('COLD')>, 'mod_coco_fpath': <Value(None)>, 'out_dpath': <Value(None)>, 'prob': <Value(0.99)>, 'resolution': <Value('30GSD')>, 'sensors': <Value('L8')>, 'timestamp': <Value(False)>, 'track_emissions': <Value(True)>, 'workermode': <Value('process')>, 'workers': <Value(16)>, 'write_kwcoco': <Value(True)>, 'year_highbound': <Value(None)>, 'year_lowbound': <Value(None)>}¶

geowatch.tasks.cold.predict.cold_predict_main(cmdline=1, **kwargs)[source]¶

Parameters:: cmdline (int, optional) – _description_. Defaults to 1.