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

Dealing with unusual image formats (16 bit and 4 channel).

Browse Source
This commit is contained in:
Tadas Baltrusaitis
2017-09-02 09:01:24 +01:00
parent 7879cbbb5e
commit c96aa6391b
5 changed files with 69 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1766,4 +1766,45 @@ void convert_to_grayscale(const cv::Mat& in, cv::Mat& out)
}
}
void convert_to_8bit_bgr_or_grayscale(cv::Mat& in_out)
{
if (in_out.channels() == 3)
{
// Make sure it's in a correct format
if (in_out.depth() != CV_8U)
{
if (in_out.depth() == CV_16U)
{
in_out = in_out / 256;
in_out.convertTo(in_out, CV_8UC3);
}
else if (in_out.depth() != CV_8U)
{
in_out.convertTo(in_out, CV_8U);
}
}
}
else if (in_out.channels() == 4)
{
cv::cvtColor(in_out, in_out, CV_BGRA2BGR);
if (in_out.depth() == CV_16U)
{
in_out = in_out / 256;
in_out.convertTo(in_out, CV_8UC3);
}
}
else
{
if (in_out.depth() == CV_16U)
{
in_out = in_out / 256;
}
else if (in_out.depth() != CV_8U)
{
in_out.convertTo(in_out, CV_8U);
}
}
}
}