Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

dvc.yaml 3.2 KB

You have to be logged in to leave a comment. Sign In
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
  1. # dvc project pipeline
  2. # NOTE: see scripts/README.MD for preprocessing from raw data up to the first
  3. # tracked stage `createtiles`
  4. stages:
  5. createtiles:
  6. foreach:
  7. - 2017
  8. - 2018
  9. - 2019
  10. - 2020
  11. do:
  12. cmd: >-
  13. mkdir -p data/processed.images.${item};
  14. gdal_retile.py
  15. -csv locations.csv
  16. -v -ps ${source_dim} ${source_dim}
  17. -co "TILED=YES" -co "COMPRESS=LZW" -co "PREDICTOR=2" -co "ALPHA=NO" -co "NUM_THREADS=ALL_CPUS"
  18. -targetDir data/processed.images.${item}
  19. data/raw/ortho_ms_${item}_EPSG3044.tif
  20. deps:
  21. - data/raw/ortho_ms_${item}_EPSG3044.tif
  22. params:
  23. - source_dim
  24. outs:
  25. - data/processed.images.${item}
  26. computestats:
  27. cmd: >-
  28. python scripts/computestats.py --frac 0.1
  29. data/processed.images.2017
  30. data/processed.images.2018
  31. data/processed.images.2019
  32. data/processed.images.2020
  33. deps:
  34. - data/processed.images.2017
  35. - data/processed.images.2018
  36. - data/processed.images.2019
  37. - data/processed.images.2020
  38. outs:
  39. - data/processed.images.stats.json
  40. createmasks:
  41. cmd: >-
  42. python scripts/createmasks.py
  43. data/processed.images.2019
  44. data/processed.masks.2019
  45. data/raw/shapefiles/deadtrees_2019/deadtrees_2019.shp
  46. --negativesample data/raw/shapefiles/deadtrees_notdead/deadtrees_notdead.shp
  47. deps:
  48. - data/processed.images.2019
  49. - data/raw/shapefiles/deadtrees_2019
  50. - data/raw/shapefiles/deadtrees_notdead
  51. outs:
  52. - data/processed.masks.2019
  53. - data/processed.masks.2019.neg_sample
  54. createdataset:
  55. cmd: >-
  56. python scripts/createdataset.py
  57. data/processed.images.2019
  58. data/processed.masks.2019
  59. data/dataset
  60. --source_dim ${source_dim}
  61. --tile_size ${createdataset.tile_size}
  62. --format ${file_type}
  63. deps:
  64. - data/processed.images.2019
  65. - data/processed.masks.2019
  66. - data/processed.masks.2019.neg_sample
  67. params:
  68. - source_dim
  69. - createdataset.tile_size
  70. - file_type
  71. outs:
  72. - data/dataset/train
  73. - data/dataset/stats.csv
  74. # train: do this manually
  75. # inference
  76. inference:
  77. foreach:
  78. - 2017
  79. - 2018
  80. - 2019
  81. - 2020
  82. do:
  83. cmd: >-
  84. mkdir -p data/predicted.${item};
  85. stdbuf -i0 -o0 -e0 python scripts/inference.py --all --nopreview -o data/predicted.${item} data/processed.images.${item};
  86. gdal_merge.py
  87. -co "TILED=YES" -co "COMPRESS=LZW" -co "PREDICTOR=2" -co "NUM_THREADS=ALL_CPUS"
  88. -o data/predicted_mosaic_${item}.tif
  89. data/predicted.${item}/ortho_ms_${item}_EPSG3044_*
  90. deps:
  91. - data/processed.images.${item}
  92. - checkpoints/bestmodel.ckpt
  93. outs:
  94. - data/predicted.${item}
  95. - data/predicted_mosaic_${item}.tif
  96. computestatsinference:
  97. cmd: >-
  98. python scripts/computestats_inference.py
  99. data/predicted.2017
  100. data/predicted.2018
  101. data/predicted.2019
  102. data/predicted.2020
  103. deps:
  104. - data/predicted.2017
  105. - data/predicted.2018
  106. - data/predicted.2019
  107. - data/predicted.2020
  108. outs:
  109. - data/predicted.stats.csv
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...