uvos/SDImagePreprocess
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Wip setream carving

Browse source
This commit is contained in:
Carl Philipp Klemm 2023-06-29 19:25:31 +02:00
parent 438c9d726c
commit f5dad284e6
11 changed files with 287 additions and 387 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
yolo.cpp
View file

@ -22,6 +22,7 @@ Yolo::Yolo(const std::filesystem::path &onnxModelPath, const cv::Size &modelInpu
if(classesTxtFilePath.empty())
{
Log(Log::INFO)<<"Using builtin classes";
loadClasses(rdefaultClassesData);
}
else
@ -31,10 +32,14 @@ Yolo::Yolo(const std::filesystem::path &onnxModelPath, const cv::Size &modelInpu
}
if(!modelPath.empty())
{
net = cv::dnn::readNetFromONNX(modelPath);
}
else
{
Log(Log::INFO)<<"Using builtin yolo model";
net = cv::dnn::readNetFromONNX((const char*)rdefaultModelData, rdefaultModelSize);
}
if(runWithOCl)
{
std::cout << "\nRunning on OCV" << std::endl;