You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
A Spatio-temporal Framework for Soil Property Prediction with Digital Soil Mapping (DSM)
3
2
4
-
This new architecture, incorporates spatial information using a base convolutional neural network (CNN) model and spatial attention mechanism, along with climate temporal information using a long short-term memory (LSTM) network.
5
-
6
-
In case you need more information, feel free to send an email: [email protected]
7
-
8
-
Paper: https://arxiv.org/abs/2308.03586
9
-
10
-
To access LUCAS topsoil dataset: https://esdac.jrc.ec.europa.eu/content/topsoil-physical-properties-europe-based-lucas-topsoil-data
11
3
12
4
<!-- ## Experiments
13
5
@@ -26,185 +18,29 @@ To access LUCAS topsoil dataset: https://esdac.jrc.ec.europa.eu/content/topsoil-
26
18
27
19
28
20
21
+
22
+
23
+
24
+
29
25
<!-- ### MODEL
30
26
- ~~Add Spatial Attention Module~~
31
27
- ~~FCNN + Regressor~~
32
28
- ~~FCNN + SA + Regressor~~
33
29
- ~~LSTM~~
34
30
-->
35
-
36
31
<!-- ### DATASET
37
32
- ~~Add TerraClimate Dataset~~
38
33
- ~~Update the ClimateInformation.js (in processing)~~
Although to train, you're gonna need to have the `.csv` files. namely LUCAS dataset under the flag of `--lucas_csv` and the TerraClimate dataset under the flag of `--climate_csv_folder_path`.
106
-
107
-
The output is a **Training Plot and** a **JSON** file containing all of the results of the **cross-validation**. all will be saved in the `results/` folder
108
-
109
-
-->
110
-
111
-
112
-
113
-
114
-
<!--
115
-
## Explanation of Command-line Flags for `train.py`
116
-
117
-
The `train.py` script accepts several command-line flags (arguments) that allow you to customize the training process. These flags help adjust various settings and parameters for the model training. Below is a detailed explanation of each flag:
118
-
119
-
1. `-nw` or `--num_workers`:
120
-
- Type: Integer
121
-
- Default: 2
122
-
- Description: Number of workers used for data loading during training. Adjust this value based on your system's capabilities to optimize data loading efficiency.
123
-
124
-
2. `-tbs` or `--train_batch_size`:
125
-
- Type: Integer
126
-
- Default: 4
127
-
- Description: Batch size used during training. A larger batch size can increase training speed but may require more memory.
128
-
129
-
3. `-Tbs` or `--test_batch_size`:
130
-
- Type: Integer
131
-
- Default: 4
132
-
- Description: Batch size used during testing. Similar to the training batch size, it affects memory consumption during testing.
133
-
134
-
4. `-lr` or `--learning_rate`:
135
-
- Type: Float
136
-
- Default: 0.0001
137
-
- Description: Learning rate used for the optimization algorithm during training.
138
-
139
-
5. `-ne` or `--num_epochs`:
140
-
- Type: Integer
141
-
- Default: 2
142
-
- Description: Number of training epochs. An epoch is a complete pass through the entire training dataset.
143
-
144
-
6. `-lrs` or `--lr_scheduler`:
145
-
- Choices: 'step', 'plateau', or None (case-sensitive)
146
-
- Default: 'step'
147
-
- Description: Learning rate scheduler type. Choose from 'step' (step-wise decay), 'plateau' (decay on validation loss plateau), or None (no learning rate decay).
148
-
149
-
7. `-oc` or `--oc_max`:
150
-
- Type: Integer
151
-
- Default: 87
152
-
- Description: Maximum value for OC during training. Adjust this value based on your specific use case.
153
-
154
-
8. `-us` or `--use_srtm`:
155
-
- Action: Store True
156
-
- Default: True
157
-
- Description: Enable or disable the use of SRTM (Shuttle Radar Topography Mission) data during training.
158
-
159
-
9. `-usa` or `--use_spatial_attention`:
160
-
- Action: Store True
161
-
- Default: True
162
-
- Description: Enable or disable the use of spatial attention in the model architecture.
163
-
164
-
10. `-ca` or `--cnn_architecture`:
165
-
- Choices: 'vgg16', 'resnet101' (case-sensitive)
166
-
- Default: 'vgg16'
167
-
- Description: Choose the CNN architecture for the model. Options are 'vgg16' or 'resnet101'.
168
-
169
-
11. `-rv` or `--reg_version`:
170
-
- Type: Integer
171
-
- Default: 2
172
-
- Description: Regression version used during training. Adjust this value based on your specific use case.
173
-
174
-
12. `-ulb` or `--use_lstm_branch`:
175
-
- Action: Store True
176
-
- Default: True
177
-
- Description: Enable or disable the use of the LSTM branch in the model architecture.
These command-line flags allow you to configure various aspects of the model training process based on your specific requirements and dataset. Adjust the values according to your needs when running the `train.py` script.
0 commit comments