ZF/OpenFace
1
0
Fork 0
mirror of https://gitcode.com/gh_mirrors/ope/OpenFace.git synced 2026-05-20 14:17:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplifying conversion to grayscale

Browse Source
This commit is contained in:
Tadas Baltrusaitis
2017-08-26 17:01:54 +01:00
parent 2ea630c834
commit f62516789c
5 changed files with 54 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
lib/local/LandmarkDetector/src/LandmarkDetectorUtils.cpp
View File

@@ -1724,4 +1724,46 @@ void SkipComments(std::ifstream& stream)
}
}
// Some other utility functions
void convert_to_grayscale(const cv::Mat& in, cv::Mat& out)
{
if (in.channels() == 3)
{
// Make sure it's in a correct format
if (in.depth() != CV_8U)
{
if (in.depth() == CV_16U)
{
cv::Mat tmp = in / 256;
tmp.convertTo(tmp, CV_8U);
cv::cvtColor(tmp, out, CV_BGR2GRAY);
}
}
else
{
cv::cvtColor(in, out, CV_BGR2GRAY);
}
}
else if (in.channels() == 4)
{
cv::cvtColor(in, out, CV_BGRA2GRAY);
}
else
{
if (in.depth() == CV_16U)
{
cv::Mat tmp = in / 256;
out = tmp.clone();
}
else if (in.depth() != CV_8U)
{
in.convertTo(out, CV_8U);
}
else
{
out = in.clone();
}
}
}
}