[pj_tflite_edge_dexined] add another sample image

iwatake2222 · Jan 14, 2022 · 929ecd7 · 929ecd7
1 parent 48e9d8f
commit 929ecd7
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/pj_tflite_edge_dexined/00_doc/edge.jpg b/pj_tflite_edge_dexined/00_doc/edge.jpg
diff --git a/pj_tflite_edge_dexined/00_doc/edge_2.jpg b/pj_tflite_edge_dexined/00_doc/edge_2.jpg
diff --git a/pj_tflite_edge_dexined/README.md b/pj_tflite_edge_dexined/README.md
@@ -3,6 +3,8 @@ Sample project to run DexiNed
 
 ![00_doc/edge.jpg](00_doc/edge.jpg)
 
+![00_doc/edge_2.jpg](00_doc/edge_2.jpg)
+
 ## How to Run
 1. Please follow the instruction: https://github.com/iwatake2222/play_with_tflite/blob/master/README.md
 2. Additional steps:
@@ -14,3 +16,4 @@ Sample project to run DexiNed
 ## Acknowledgements
 - https://github.com/xavysp/DexiNed
 - https://github.com/PINTO0309/PINTO_model_zoo
+- https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/resources.html
diff --git a/pj_tflite_edge_dexined/image_processor/image_processor.cpp b/pj_tflite_edge_dexined/image_processor/image_processor.cpp
@@ -115,8 +115,9 @@ int32_t ImageProcessor::Process(cv::Mat& mat, ImageProcessor::Result& result)
     }
 
     /* Convert to colored image */
-    cv::Mat mat_edge;
-    cv::cvtColor(engine_result.mat_out, mat_edge, cv::COLOR_GRAY2BGR);
+    cv::Mat mat_edge = engine_result.mat_out;
+    //cv::threshold(engine_result.mat_out, mat_edge, 128, 255, cv::THRESH_BINARY);
+    cv::cvtColor(mat_edge, mat_edge, cv::COLOR_GRAY2BGR);
 
     /* Create result image */
     cv::resize(mat_edge, mat_edge, mat.size());