From e1119953ac979628adb2e50560f46da54079fb32 Mon Sep 17 00:00:00 2001 From: JC_Xiong Date: Thu, 25 Apr 2024 10:22:55 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Code/MF_TRTInfer/utils.cpp | 42 +++++++++++++++++++------------------- Code/MF_TRTInfer/utils.h | 2 +- 2 files changed, 22 insertions(+), 22 deletions(-) diff --git a/Code/MF_TRTInfer/utils.cpp b/Code/MF_TRTInfer/utils.cpp index f52d087..3f82e0d 100644 --- a/Code/MF_TRTInfer/utils.cpp +++ b/Code/MF_TRTInfer/utils.cpp @@ -1,6 +1,6 @@ #include "utils.h" -void utils::saveBinaryFile(float* vec, size_t len, const std::string& file) +void trtUtils::saveBinaryFile(float* vec, size_t len, const std::string& file) { std::ofstream out(file, std::ios::out | std::ios::binary); if (!out.is_open()) @@ -9,7 +9,7 @@ void utils::saveBinaryFile(float* vec, size_t len, const std::string& file) out.close(); } -std::vector utils::readBinaryFile(const std::string& file) +std::vector trtUtils::readBinaryFile(const std::string& file) { std::ifstream in(file, std::ios::in | std::ios::binary); if (!in.is_open()) @@ -30,7 +30,7 @@ std::vector utils::readBinaryFile(const std::string& file) } -std::vector utils::loadModel(const std::string& file) +std::vector trtUtils::loadModel(const std::string& file) { std::ifstream in(file, std::ios::in | std::ios::binary); if (!in.is_open()) @@ -51,19 +51,19 @@ std::vector utils::loadModel(const std::string& file) return data; } -std::string utils::getSystemTimeStr() +std::string trtUtils::getSystemTimeStr() { return std::to_string(std::rand()); } -bool utils::setInputStream(const utils::InputStream& source, const std::string& imagePath, const std::string& videoPath, const int& cameraID, - cv::VideoCapture& capture, int& totalBatches, int& delayTime, utils::InitParameter& param) +bool trtUtils::setInputStream(const trtUtils::InputStream& source, const std::string& imagePath, const std::string& videoPath, const int& cameraID, + cv::VideoCapture& capture, int& totalBatches, int& delayTime, trtUtils::InitParameter& param) { int total_frames = 0; std::string img_format; switch (source) { - case utils::InputStream::IMAGE: + case trtUtils::InputStream::IMAGE: img_format = imagePath.substr(imagePath.size()-4, 4); if (img_format == ".png" || img_format == ".PNG") { @@ -79,13 +79,13 @@ bool utils::setInputStream(const utils::InputStream& source, const std::string& totalBatches = 1; delayTime = 0; break; - case utils::InputStream::VIDEO: + case trtUtils::InputStream::VIDEO: capture.open(videoPath); total_frames = capture.get(cv::CAP_PROP_FRAME_COUNT); totalBatches = (total_frames % param.batch_size == 0) ? (total_frames / param.batch_size) : (total_frames / param.batch_size + 1); break; - case utils::InputStream::CAMERA: + case trtUtils::InputStream::CAMERA: capture.open(cameraID); total_frames = INT_MAX; totalBatches = INT_MAX; @@ -99,7 +99,7 @@ bool utils::setInputStream(const utils::InputStream& source, const std::string& return capture.isOpened(); } -void utils::setRenderWindow(InitParameter& param) +void trtUtils::setRenderWindow(InitParameter& param) { if (!param.is_show) return; @@ -122,14 +122,14 @@ void utils::setRenderWindow(InitParameter& param) } } -std::string utils::getTimeStamp() +std::string trtUtils::getTimeStamp() { std::chrono::nanoseconds t = std::chrono::duration_cast( std::chrono::system_clock::now().time_since_epoch()); return std::to_string(t.count()); } -void utils::show(const std::vector>& objectss, const std::vector& classNames, +void trtUtils::show(const std::vector>& objectss, const std::vector& classNames, const int& cvDelayTime, std::vector& imgsBatch) { std::string windows_title = "image"; @@ -191,7 +191,7 @@ void utils::show(const std::vector>& objectss, const std } } -void utils::save(const std::vector>& objectss, const std::vector& classNames, +void trtUtils::save(const std::vector>& objectss, const std::vector& classNames, const std::string& savePath, std::vector& imgsBatch, const int& batchSize, const int& batchi) { cv::Scalar color = cv::Scalar(0, 255, 0); @@ -241,30 +241,30 @@ void utils::save(const std::vector>& objectss, const std::vecto } } -utils::HostTimer::HostTimer() +trtUtils::HostTimer::HostTimer() { t1 = std::chrono::steady_clock::now(); } -float utils::HostTimer::getUsedTime() +float trtUtils::HostTimer::getUsedTime() { t2 = std::chrono::steady_clock::now(); std::chrono::duration time_used = std::chrono::duration_cast>(t2 - t1); return(1000 * time_used.count()); // ms } -utils::HostTimer::~HostTimer() +trtUtils::HostTimer::~HostTimer() { } -utils::DeviceTimer::DeviceTimer() +trtUtils::DeviceTimer::DeviceTimer() { cudaEventCreate(&start); cudaEventCreate(&end); cudaEventRecord(start); } -float utils::DeviceTimer::getUsedTime() +float trtUtils::DeviceTimer::getUsedTime() { cudaEventRecord(end); cudaEventSynchronize(end); @@ -273,14 +273,14 @@ float utils::DeviceTimer::getUsedTime() return total_time; } -utils::DeviceTimer::DeviceTimer(cudaStream_t stream) +trtUtils::DeviceTimer::DeviceTimer(cudaStream_t stream) { cudaEventCreate(&start); cudaEventCreate(&end); cudaEventRecord(start, stream); } -float utils::DeviceTimer::getUsedTime(cudaStream_t stream) +float trtUtils::DeviceTimer::getUsedTime(cudaStream_t stream) { cudaEventRecord(end, stream); cudaEventSynchronize(end); @@ -289,7 +289,7 @@ float utils::DeviceTimer::getUsedTime(cudaStream_t stream) return total_time; } -utils::DeviceTimer::~DeviceTimer() +trtUtils::DeviceTimer::~DeviceTimer() { cudaEventDestroy(start); cudaEventDestroy(end); diff --git a/Code/MF_TRTInfer/utils.h b/Code/MF_TRTInfer/utils.h index 5b3dea6..a21b952 100644 --- a/Code/MF_TRTInfer/utils.h +++ b/Code/MF_TRTInfer/utils.h @@ -14,7 +14,7 @@ #endif -namespace utils +namespace trtUtils { namespace dataSets {