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
Copy file name to clipboardExpand all lines: README.md
+2-1Lines changed: 2 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -17,6 +17,7 @@ Original Image |Mask generated by the model|Mask overlaid on origina
17
17
18
18
### About
19
19
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.
20
21
21
22
### Massachusetts Roads Dataset Summary
22
23
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
42
43
./setup.sh
43
44
```
44
45
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.
46
47
47
48
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.
0 commit comments