tensorflow object detection api目標檢測之設置顯示的閾值

來自專欄 機器學習

設置顯示的閾值，也就是說當檢測的分數小於一定值時就不顯示檢測的目標

想設置閾值，各種搜索找了好久，都沒有找到我這個問題的答案，所以乾脆自己想辦法解決，解決問題的辦法是閱讀代碼，毫無疑問，添加檢測框的代碼就是測試代碼中的這一行代碼，如下圖

上圖所圈處就是添加檢測框的代碼，這個代碼用到了vis_util模塊的visualize_boxes_and_labels_on_image_array()函數，而vis_util模塊其實就是visualization_utils的別名，如下圖

而visualization_utils模塊來自於utils，所以到此已經找到需要修改的文件了，就在

researchobject_detectionutils下的visualization_utils模塊，打開visualization_utils文件，搜索定義visualize_boxes_and_labels_on_image_array()函數的位置，如下圖

上圖所圈處的min_score_thresh參數就是設置顯示的閾值，也就是代碼是設置的顯示50%的，那麼我們想設置高點，就只需修改這個這個值即可，我這裡設置為0.8，如下圖

到此為止，設置顯示的閾值的工作已經完成，去運行測試程序會看到只顯示大於或等於80%的檢測分數

推薦閱讀：