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

Add tiny yolo 512 test

Browse Source 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
xavier
2020-03-10 15:40:11 +01:00
parent 3eb079dd13
commit 2d46f16b70
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+3
View File
@@ -88,6 +88,9 @@ target_link_libraries(test_yolo3 tkDNN)
add_executable(test_yolo3_tiny tests/yolo3_tiny/yolo3_tiny.cpp) add_executable(test_yolo3_tiny tests/yolo3_tiny/yolo3_tiny.cpp)
target_link_libraries(test_yolo3_tiny tkDNN) target_link_libraries(test_yolo3_tiny tkDNN)
add_executable(test_yolo3_tiny512 tests/yolo3_tiny512/yolo3_tiny512.cpp)
target_link_libraries(test_yolo3_tiny512 tkDNN)
add_executable(test_yolo3_berkeley tests/yolo3_berkeley/yolo3_berkeley.cpp) add_executable(test_yolo3_berkeley tests/yolo3_berkeley/yolo3_berkeley.cpp)
target_link_libraries(test_yolo3_berkeley tkDNN) target_link_libraries(test_yolo3_berkeley tkDNN)
src/CenternetDetection.cpp
+8 -3
View File
@@ -1,4 +1,9 @@
#include "CenternetDetection.h" #include "CenternetDetection.h"
#include "opencv2/imgproc/imgproc.hpp"
#include <opencv2/cudawarping.hpp>
#include <opencv2/cudaarithm.hpp>
namespace tk { namespace dnn { namespace tk { namespace dnn {
@@ -226,7 +231,7 @@ void CenternetDetection::update(cv::Mat &imageORIG) {
sz_old = sz; sz_old = sz;
cv::cuda::GpuMat im_Orig; cv::cuda::GpuMat im_Orig;
im_Orig = cv::cuda::GpuMat(imageORIG); im_Orig = cv::cuda::GpuMat(imageORIG);
// cv::cuda::resize (im_Orig, imageF1_d, cv::Size(new_width, new_height)); cv::cuda::resize (im_Orig, imageF1_d, cv::Size(new_width, new_height));
checkCuda( cudaDeviceSynchronize() ); checkCuda( cudaDeviceSynchronize() );
sz = imageF1_d.size(); sz = imageF1_d.size();
@@ -235,7 +240,7 @@ void CenternetDetection::update(cv::Mat &imageORIG) {
std::cout << " TIME resize: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl; std::cout << " TIME resize: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl;
step_t = end_t; step_t = end_t;
// cv::cuda::warpAffine(imageF1_d, imageF2_d, trans, cv::Size(inp_width, inp_height), cv::INTER_LINEAR ); cv::cuda::warpAffine(imageF1_d, imageF2_d, trans, cv::Size(inp_width, inp_height), cv::INTER_LINEAR );
checkCuda( cudaDeviceSynchronize() ); checkCuda( cudaDeviceSynchronize() );
end_t = std::chrono::steady_clock::now(); end_t = std::chrono::steady_clock::now();
std::cout << " TIME warpAffine: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl; std::cout << " TIME warpAffine: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl;
@@ -248,7 +253,7 @@ void CenternetDetection::update(cv::Mat &imageORIG) {
step_t = end_t; step_t = end_t;
dim2 = dim; dim2 = dim;
// cv::cuda::split(imageF1_d,bgr);//split source cv::cuda::split(imageF1_d,bgr);//split source
end_t = std::chrono::steady_clock::now(); end_t = std::chrono::steady_clock::now();
std::cout << " TIME split: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl; std::cout << " TIME split: " << std::chrono::duration_cast<std::chrono:: microseconds>(end_t - step_t).count() << " us" << std::endl;
step_t = end_t; step_t = end_t;