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

Add colormap for each segmentation dataset

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2021-07-20 15:29:26 +02:00
parent 84ff978ccb
commit 69256992a3
6 changed files with 308 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/seg_demo.cpp
+1 -10
View File
@@ -66,15 +66,6 @@ int main(int argc, char *argv[]) {
if(n_batch < 1 || n_batch > 64)
FatalError("Batch dim not supported");
std::string net_name;
removePathAndExtension(net, net_name);
bool mapillary_15_colormap = false;
if( n_classes == 15 &&
( net_name == "shelfnet_mapillary_fp32" ||
net_name == "shelfnet_mapillary_fp16" ||
net_name == "shelfnet_mapillary_int8" ) )
mapillary_15_colormap = true;
//net initialization
tk::dnn::SegmentationNN segNN;
segNN.init(net, n_classes, n_batch);
@@ -130,7 +121,7 @@ int main(int argc, char *argv[]) {
width = frame.cols;
//inference
segNN.updateOriginal(frame, true, mapillary_15_colormap);
segNN.updateOriginal(frame, true);
if(show)
segNN.draw();