Skip to content

Commit a3a30d7

Browse files
Update README.md
1 parent c2c27a3 commit a3a30d7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ Original Image |Mask generated by the model|Mask overlaid on origina
1717

1818
### About
1919
In this work, we implement the U-Net segmentation architecture on the Mnih et. al. Massachusetts Roads Dataset for the task of road network extraction. The trained model achieves a mask accuracy of 96% on the test set. The model was trained on AWS P2.x instance.
20+
Inference time on a single K80 Tesla GPU(AWS P2.x instance) is 1.5 second and on CPU is 6 seconds.
2021

2122
### Massachusetts Roads Dataset Summary
2223
The Massachusetts Roads Dataset by Mnih et. al. is freely available at [here](https://www.cs.toronto.edu/~vmnih/data/). There is also a torrent link available [here](http://academictorrents.com/details/3b17f08ed5027ea24db04f460b7894d913f86c21).
@@ -42,7 +43,7 @@ chmod 777 setup.sh
4243
./setup.sh
4344
```
4445

45-
To train the model, proceed with step 3. To test the pretrained model, skip to [Quick start](#quick-start) step 2.
46+
To train the model, proceed with step 3 and then [Quick start](#quick-start) step 1. To test the pretrained model, simply skip to [Quick start](#quick-start) step 2.
4647

4748
3. Download the dataset from [here](http://academictorrents.com/details/3b17f08ed5027ea24db04f460b7894d913f86c21) or [here](https://www.cs.toronto.edu/~vmnih/data/). It is recommended to download the dataset using the torrent link since it downloads the files in the appropriate directories.
4849

0 commit comments

Comments
 (0)