Skip to content

Commit 59b45dc

Browse files
Add README.md
1 parent 6264d8d commit 59b45dc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,8 @@ You have 2 options
5252
1. Train: To train the model on the dataset, make sure to download the dataset and follow the repo structure as:
5353

5454

55-
```
5655
The repo structure should be as follows:
57-
```
56+
```angular2html
5857
RoadNetworkExtraction-MoveHack
5958
|_ mass_roads/
6059
| |_ train/
@@ -91,6 +90,7 @@ RoadNetworkExtraction-MoveHack
9190
|_ ....
9291
|_ (other files)
9392
```
93+
9494
Now, start training with the following command- (NOTE: This will first set up the necessary folders and convert .tiff files to png and save them. Then it will start trained the u-net model for num_epochs(default is 1) and with cycle_len(default is 4). The trained model will be saved to the models/ directory. The trained model achieves a mask accuracy of 96% on test set.)
9595

9696
```bash

0 commit comments

Comments
 (0)