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.0 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
  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. - 2019
  80. do:
  81. cmd: >-
  82. mkdir -p data/predicted.${item};
  83. stdbuf -i0 -o0 -e0 python scripts/inference.py --all --nopreview -o data/predicted.${item} data/processed.images.${item};
  84. gdal_merge.py
  85. -co "TILED=YES" -co "COMPRESS=LZW" -co "PREDICTOR=2" -co "NUM_THREADS=ALL_CPUS"
  86. -o data/predicted_mosaic_${item}.tif
  87. data/predicted.${item}/ortho_ms_${item}_EPSG3044_*
  88. deps:
  89. - data/processed.images.${item}
  90. - checkpoints/bestmodel.ckpt
  91. outs:
  92. - data/predicted.${item}
  93. - data/predicted_mosaic_${item}.tif
  94. computestatsinference:
  95. cmd: >-
  96. python scripts/computestats_inference.py
  97. data/predicted.2017
  98. data/predicted.2019
  99. deps:
  100. - data/predicted.2017
  101. - data/predicted.2019
  102. outs:
  103. - data/predicted.stats.csv
Tip!

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

Comments

Loading...