pull/10/head
triple-Mu 2 years ago
parent 69378be8b4
commit 6d69865fe2
  1. 6
      csrc/segment/CMakeLists.txt
  2. 12
      docs/Segment.md

@ -53,3 +53,9 @@ add_executable(${PROJECT_NAME}
target_link_directories(${PROJECT_NAME} PUBLIC ${ALL_LIBS})
target_link_libraries(${PROJECT_NAME} PRIVATE nvinfer nvinfer_plugin cudart ${OpenCV_LIBS})
if(${OpenCV_VERSION} VERSION_GREATER_EQUAL 4.7.0)
message(STATUS "Build with -DBATCHED_NMS")
add_definitions(-DBATCHED_NMS)
endif()

@ -112,16 +112,12 @@ mv yolov8-seg ${root}
cd ${root}
```
If you have build OpenCV(>=4.7.0) by yourself, it provides a new api [`cv::dnn::NMSBoxesBatched`](https://docs.opencv.org/4.x/d6/d0f/group__dnn.html#ga977aae09fbf7c804e003cfea1d4e928c) .
It is a gread api about efficient in-class nms .
***Notice !!!***
You can build by `-DBATCHED_NMS` and try to use it .
``` shell
....
cmake .. -DBATCHED_NMS
....
```
If you have build OpenCV(>=4.7.0) by yourself, it provides a new api [`cv::dnn::NMSBoxesBatched`](https://docs.opencv.org/4.x/d6/d0f/group__dnn.html#ga977aae09fbf7c804e003cfea1d4e928c) .
It is a gread api about efficient in-class nms . It will be used by default!
***!!!***
Usage:

Loading…
Cancel
Save