-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathdvc.yaml
54 lines (54 loc) · 2.33 KB
/
dvc.yaml
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
stages:
merge_bands:
cmd: python src/preparedata/merge_bands.py data/raw_data/tiles -s 29/S/MC -s 29/S/MD
-s 29/S/NB -s 29/S/NC -s 29/S/ND -s 29/S/PB -s 29/S/PC -s 29/S/PD -s 29/T/ME
-s 29/T/NE -s 29/T/NF -s 29/T/NG -s 29/T/PE -s 29/T/PF data/processed_data/scenes
deps:
- src/preparedata/merge_bands.py
outs:
- data/processed_data/scenes/
extract_patches:
cmd: python -m src.preparedata.extract_patches data/raw_data/wildfires-ground-truth/portugal/AArdida2016_ETRS89PTTM06_20190813.shp
data/processed_data/extracted DHFim data/processed_data/scenes -s 29/S/MC -s
29/S/MD -s 29/S/NB -s 29/S/NC -s 29/S/ND -s 29/S/PB -s 29/S/PC -s 29/S/PD -s
29/T/ME -s 29/T/NE -s 29/T/NF -s 29/T/NG -s 29/T/PE -s 29/T/PF
deps:
- data/processed_data/scenes/
- data/raw_data/wildfires-ground-truth/portugal/AArdida2016_ETRS89PTTM06_20190813.shp
- src/preparedata/extract_patches.py
- src/preparedata/patching.py
- src/preparedata/patchutils
outs:
- data/processed_data/extracted/
sort_patches:
cmd: python -m src.preparedata.sort_patches data/processed_data/training_patches
data/processed_data/extracted/patches data/processed_data/extracted/annotations
deps:
- data/processed_data/extracted/
- src/preparedata/sort_patches.py
outs:
- data/processed_data/training_patches/annotations/
- data/processed_data/training_patches/patches/
split_data:
cmd: python -m src.preparedata.split --root=data/processed_data/training_patches/
deps:
- data/processed_data/training_patches/annotations/
- data/processed_data/training_patches/patches/
- src/preparedata/split.py
outs:
- data/processed_data/training_patches/training_indices.json
resnetunet:
cmd: python -m src.run_scripts.run_training --seed=11 --gpu=1 --save-images=1
--baseline=1 --model-name=resnetunet --group=resnetunet --save-path=data/results/
deps:
- data/processed_data/training_patches/annotations/
- data/processed_data/training_patches/patches/
- data/processed_data/training_patches/training_indices.json
- src/run_scripts/run_training.py
- src/segmentation/data/
- src/segmentation/evaluation/
- src/segmentation/models/resnetunet/
- src/segmentation/networks/resnetunet/
outs:
- data/results/
- checkpoint/resnetunet/