crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Mixed updates and improvements.

Browse Source
This commit is contained in:
Daniel
2025-01-21 16:23:19 +01:00
parent c5f190ab35
commit 29c072400f
11 changed files with 49 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
scripts/utils_2d_pose.hpp
View File

@ -1,13 +1,13 @@
#pragma once
#include <filesystem>
#include <vector>
#include <string>
#include <memory>
#include <string>
#include <vector>
#include <opencv2/opencv.hpp>
#include <onnxruntime_cxx_api.h>
#include <onnxruntime_c_api.h>
#include <opencv2/opencv.hpp>
#include <tensorrt_provider_options.h>
// =================================================================================================
@ -646,27 +646,10 @@ namespace utils_2d_pose
std::vector<std::array<float, 5>> RTMDet::call(const cv::Mat &image)
{
auto stime = std::chrono::high_resolution_clock::now();
std::chrono::duration<float> elapsed;
cv::Mat preprocessed = preprocess(image);
std::vector<cv::Mat> inputs = {preprocessed};
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Preprocess time: " << elapsed.count() << "s\n";
stime = std::chrono::high_resolution_clock::now();
auto results = call_by_image(inputs);
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Inference time: " << elapsed.count() << "s\n";
stime = std::chrono::high_resolution_clock::now();
auto outputs = postprocess(results, image);
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Postprocess time: " << elapsed.count() << "s\n";
return outputs;
}
@ -818,26 +801,9 @@ namespace utils_2d_pose
std::vector<std::vector<std::array<float, 3>>> RTMPose::call(
const cv::Mat &image, const std::vector<std::array<float, 5>> &bboxes)
{
auto stime = std::chrono::high_resolution_clock::now();
std::chrono::duration<float> elapsed;
std::vector<cv::Mat> inputs = preprocess(image, bboxes);
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Preprocess time: " << elapsed.count() << "s\n";
stime = std::chrono::high_resolution_clock::now();
auto results = call_by_image(inputs);
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Inference time: " << elapsed.count() << "s\n";
stime = std::chrono::high_resolution_clock::now();
auto outputs = postprocess(results, image, bboxes);
elapsed = std::chrono::high_resolution_clock::now() - stime;
std::cout << "Postprocess time: " << elapsed.count() << "s\n";
return outputs;
}