Update README_seg

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-11-24 12:39:37 +01:00
parent a52e18b6e6
commit f137bcb694
+1 -1
View File
@@ -3,7 +3,7 @@
Currently tkDNN supports only ShelfNet as semantic segmentation network. Currently tkDNN supports only ShelfNet as semantic segmentation network.
## Export weights from Shelfnet ## Export weights from Shelfnet
To get the weights needed to run Mobilenet tests use [this](https://git.hipert.unimore.it/mverucchi/shelfnet) fork of a Pytorch implementation of Shelfnet network. To get the weights needed to run Shelfnet tests use [this](https://git.hipert.unimore.it/mverucchi/shelfnet) fork of a Pytorch implementation of Shelfnet network.
``` ```
git clone https://git.hipert.unimore.it/mverucchi/shelfnet git clone https://git.hipert.unimore.it/mverucchi/shelfnet