mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Update README_seg.md

Browse Source
This commit is contained in:
Micaela Verucchi
2020-11-24 12:44:36 +01:00
committed by GitHub
parent f137bcb694
commit 43d213ba36
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README_seg.md
+2 -2
View File
@@ -23,7 +23,7 @@ To run the semantic segmentation demo follow these steps (example with shelfnet_
rm shelfnet_mapillary_fp32.rt # be sure to delete(or move) old tensorRT files 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) ./test_shelfnet_mapillary # run the yolo test (is slow)
./demo shelfnet_mapillary_fp32.rt ../demo/yolo_test.mp4 1 15 ./seg_demo shelfnet_mapillary_fp32.rt ../demo/yolo_test.mp4 1 15
``` ```
In general the demo program takes the following parameters: In general the demo program takes the following parameters:
``` ```
@@ -53,4 +53,4 @@ N.b. By default it is used FP32 inference
1. Zhuang, Juntang, et al. "ShelfNet for fast semantic segmentation." Proceedings of the IEEE International Conference on Computer Vision Workshops. 2019. 1. Zhuang, Juntang, et al. "ShelfNet for fast semantic segmentation." Proceedings of the IEEE International Conference on Computer Vision Workshops. 2019.
*. Mapillary Vistas has originally 66 classes, but we reduced them to 15 to improve the results on the categories of our interest. *. Mapillary Vistas has originally 66 classes, but we reduced them to 15 to improve the results on the categories of our interest.