From f137bcb694baa81e41ef559344b00b611d819c1d Mon Sep 17 00:00:00 2001 From: Micaela Verucchi Date: Tue, 24 Nov 2020 12:39:37 +0100 Subject: [PATCH] Update README_seg Signed-off-by: Micaela Verucchi --- README_seg.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README_seg.md b/README_seg.md index a9afcd0..6717f0d 100644 --- a/README_seg.md +++ b/README_seg.md @@ -3,7 +3,7 @@ Currently tkDNN supports only ShelfNet as semantic segmentation network. ## 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 @@ -21,7 +21,7 @@ python export.py To run the semantic segmentation demo follow these steps (example with shelfnet_mapillary): ``` rm shelfnet_mapillary_fp32.rt # be sure to delete(or move) old tensorRT files -export TKDNN_BATCHSIZE=4 # be sure you have batch size > than 1 if you want to run inference on images bigger than 1024 +export TKDNN_BATCHSIZE=4 # be sure you have batch size > than 1 if you want to run inference on images bigger than 1024 ./test_shelfnet_mapillary # run the yolo test (is slow) ./demo shelfnet_mapillary_fp32.rt ../demo/yolo_test.mp4 1 15 ```