Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
O
opencv
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
opencv
Commits
470f6faf
Commit
470f6faf
authored
Mar 31, 2012
by
Andrey Kamaev
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fixed conversions from YV12 and IYUV on non-continuous input. Added accuracy and performance tests.
parent
a22641aa
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
544 additions
and
54 deletions
+544
-54
perf_cvt_color.cpp
modules/imgproc/perf/perf_cvt_color.cpp
+14
-7
color.cpp
modules/imgproc/src/color.cpp
+43
-47
test_cvtyuv.cpp
modules/imgproc/test/test_cvtyuv.cpp
+487
-0
No files found.
modules/imgproc/perf/perf_cvt_color.cpp
View file @
470f6faf
...
@@ -106,7 +106,9 @@ CV_ENUM(CvtMode,
...
@@ -106,7 +106,9 @@ CV_ENUM(CvtMode,
CV_YUV2BGR
,
CV_YUV2RGB
,
CX_YUV2BGRA
,
CX_YUV2RGBA
CV_YUV2BGR
,
CV_YUV2RGB
,
CX_YUV2BGRA
,
CX_YUV2RGBA
)
)
CV_ENUM
(
CvtMode2
,
CV_YUV2BGR_NV12
,
CV_YUV2BGRA_NV12
,
CV_YUV2RGB_NV12
,
CV_YUV2RGBA_NV12
,
CV_YUV420sp2BGR
,
CV_YUV420sp2BGRA
,
CV_YUV420sp2RGB
,
CV_YUV420sp2RGBA
)
CV_ENUM
(
CvtMode2
,
CV_YUV2BGR_NV12
,
CV_YUV2BGRA_NV12
,
CV_YUV2RGB_NV12
,
CV_YUV2RGBA_NV12
,
CV_YUV2BGR_NV21
,
CV_YUV2BGRA_NV21
,
CV_YUV2RGB_NV21
,
CV_YUV2RGBA_NV21
,
CV_YUV2BGR_YV12
,
CV_YUV2BGRA_YV12
,
CV_YUV2RGB_YV12
,
CV_YUV2RGBA_YV12
,
CV_YUV2BGR_IYUV
,
CV_YUV2BGRA_IYUV
,
CV_YUV2RGB_IYUV
,
CV_YUV2RGBA_IYUV
,
COLOR_YUV2GRAY_420
)
struct
ChPair
struct
ChPair
{
{
...
@@ -120,6 +122,7 @@ ChPair getConversionInfo(int cvtMode)
...
@@ -120,6 +122,7 @@ ChPair getConversionInfo(int cvtMode)
{
{
case
CV_BayerBG2GRAY
:
case
CV_BayerGB2GRAY
:
case
CV_BayerBG2GRAY
:
case
CV_BayerGB2GRAY
:
case
CV_BayerGR2GRAY
:
case
CV_BayerRG2GRAY
:
case
CV_BayerGR2GRAY
:
case
CV_BayerRG2GRAY
:
case
CV_YUV2GRAY_420
:
return
ChPair
(
1
,
1
);
return
ChPair
(
1
,
1
);
case
CV_GRAY2BGR555
:
case
CV_GRAY2BGR565
:
case
CV_GRAY2BGR555
:
case
CV_GRAY2BGR565
:
return
ChPair
(
1
,
2
);
return
ChPair
(
1
,
2
);
...
@@ -127,13 +130,17 @@ ChPair getConversionInfo(int cvtMode)
...
@@ -127,13 +130,17 @@ ChPair getConversionInfo(int cvtMode)
case
CV_BayerGB2BGR
:
case
CV_BayerGB2BGR_VNG
:
case
CV_BayerGB2BGR
:
case
CV_BayerGB2BGR_VNG
:
case
CV_BayerGR2BGR
:
case
CV_BayerGR2BGR_VNG
:
case
CV_BayerGR2BGR
:
case
CV_BayerGR2BGR_VNG
:
case
CV_BayerRG2BGR
:
case
CV_BayerRG2BGR_VNG
:
case
CV_BayerRG2BGR
:
case
CV_BayerRG2BGR_VNG
:
case
CV_GRAY2BGR
:
case
CV_YUV2BGR_NV12
:
case
CV_GRAY2BGR
:
case
CV_YUV2RGB_NV12
:
case
CV_YUV420sp2BGR
:
case
CV_YUV2BGR_NV12
:
case
CV_YUV2RGB_NV12
:
case
CV_YUV420sp2RGB
:
case
CV_YUV2BGR_NV21
:
case
CV_YUV2RGB_NV21
:
case
CV_YUV2BGR_YV12
:
case
CV_YUV2RGB_YV12
:
case
CV_YUV2BGR_IYUV
:
case
CV_YUV2RGB_IYUV
:
return
ChPair
(
1
,
3
);
return
ChPair
(
1
,
3
);
case
CV_GRAY2BGRA
:
case
CV_YUV2BGRA_NV12
:
case
CV_GRAY2BGRA
:
case
CV_YUV2RGBA_NV12
:
case
CV_YUV420sp2BGRA
:
case
CV_YUV2BGRA_NV12
:
case
CV_YUV2RGBA_NV12
:
case
CV_YUV420sp2RGBA
:
case
CV_YUV2BGRA_NV21
:
case
CV_YUV2RGBA_NV21
:
case
CV_YUV2BGRA_YV12
:
case
CV_YUV2RGBA_YV12
:
case
CV_YUV2BGRA_IYUV
:
case
CV_YUV2RGBA_IYUV
:
return
ChPair
(
1
,
4
);
return
ChPair
(
1
,
4
);
case
CV_BGR5552GRAY
:
case
CV_BGR5652GRAY
:
case
CV_BGR5552GRAY
:
case
CV_BGR5652GRAY
:
return
ChPair
(
2
,
1
);
return
ChPair
(
2
,
1
);
...
...
modules/imgproc/src/color.cpp
View file @
470f6faf
...
@@ -2829,22 +2829,18 @@ struct YUV420p2RGB888Invoker
...
@@ -2829,22 +2829,18 @@ struct YUV420p2RGB888Invoker
Mat
*
dst
;
Mat
*
dst
;
const
uchar
*
my1
,
*
mu
,
*
mv
;
const
uchar
*
my1
,
*
mu
,
*
mv
;
int
width
,
stride
;
int
width
,
stride
;
int
ustepIdx
,
vstepIdx
;
YUV420p2RGB888Invoker
(
Mat
*
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
)
YUV420p2RGB888Invoker
(
Mat
*
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
,
int
_ustepIdx
,
int
_vstepIdx
)
:
dst
(
_dst
),
my1
(
_y1
),
mu
(
_u
),
mv
(
_v
),
width
(
_dst
->
cols
),
stride
(
_stride
)
{}
:
dst
(
_dst
),
my1
(
_y1
),
mu
(
_u
),
mv
(
_v
),
width
(
_dst
->
cols
),
stride
(
_stride
)
,
ustepIdx
(
_ustepIdx
),
vstepIdx
(
_vstepIdx
)
{}
void
operator
()(
const
BlockedRange
&
range
)
const
void
operator
()(
const
BlockedRange
&
range
)
const
{
{
const
int
rangeBegin
=
range
.
begin
()
*
2
;
const
int
rangeBegin
=
range
.
begin
()
*
2
;
const
int
rangeEnd
=
range
.
end
()
*
2
;
const
int
rangeEnd
=
range
.
end
()
*
2
;
//R = 1.164(Y - 16) + 1.596(V - 128)
size_t
uvsteps
[
2
]
=
{
width
/
2
,
stride
-
width
/
2
};
//G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
int
usIdx
=
ustepIdx
,
vsIdx
=
vstepIdx
;
//B = 1.164(Y - 16) + 2.018(U - 128)
//R = (1220542(Y - 16) + 1673527(V - 128) + (1 << 19)) >> 20
//G = (1220542(Y - 16) - 852492(V - 128) - 409993(U - 128) + (1 << 19)) >> 20
//B = (1220542(Y - 16) + 2116026(U - 128) + (1 << 19)) >> 20
const
int
cY
=
1220542
;
const
int
cY
=
1220542
;
const
int
cUB
=
2116026
;
const
int
cUB
=
2116026
;
...
@@ -2854,10 +2850,16 @@ struct YUV420p2RGB888Invoker
...
@@ -2854,10 +2850,16 @@ struct YUV420p2RGB888Invoker
const
int
YUV420_SHIFT
=
20
;
const
int
YUV420_SHIFT
=
20
;
const
uchar
*
y1
=
my1
+
rangeBegin
*
stride
;
const
uchar
*
y1
=
my1
+
rangeBegin
*
stride
;
const
uchar
*
u1
=
mu
+
(
range
.
begin
()
/
2
)
*
stride
+
(
range
.
begin
()
%
2
)
*
width
/
2
;
const
uchar
*
u1
=
mu
+
(
range
.
begin
()
/
2
)
*
stride
;
const
uchar
*
v1
=
mv
+
(
range
.
begin
()
/
2
)
*
stride
+
(
range
.
begin
()
%
2
)
*
width
/
2
;
const
uchar
*
v1
=
mv
+
(
range
.
begin
()
/
2
)
*
stride
;
for
(
int
j
=
rangeBegin
;
j
<
rangeEnd
;
j
+=
2
,
y1
+=
stride
*
2
,
u1
+=
width
/
2
,
v1
+=
width
/
2
)
if
(
range
.
begin
()
%
2
==
1
)
{
u1
+=
uvsteps
[(
usIdx
++
)
&
1
];
v1
+=
uvsteps
[(
vsIdx
++
)
&
1
];
}
for
(
int
j
=
rangeBegin
;
j
<
rangeEnd
;
j
+=
2
,
y1
+=
stride
*
2
,
u1
+=
uvsteps
[(
usIdx
++
)
&
1
],
v1
+=
uvsteps
[(
vsIdx
++
)
&
1
])
{
{
uchar
*
row1
=
dst
->
ptr
<
uchar
>
(
j
);
uchar
*
row1
=
dst
->
ptr
<
uchar
>
(
j
);
uchar
*
row2
=
dst
->
ptr
<
uchar
>
(
j
+
1
);
uchar
*
row2
=
dst
->
ptr
<
uchar
>
(
j
+
1
);
...
@@ -2892,12 +2894,6 @@ struct YUV420p2RGB888Invoker
...
@@ -2892,12 +2894,6 @@ struct YUV420p2RGB888Invoker
row2
[
4
]
=
saturate_cast
<
uchar
>
((
y11
+
guv
)
>>
YUV420_SHIFT
);
row2
[
4
]
=
saturate_cast
<
uchar
>
((
y11
+
guv
)
>>
YUV420_SHIFT
);
row2
[
3
+
bIdx
]
=
saturate_cast
<
uchar
>
((
y11
+
buv
)
>>
YUV420_SHIFT
);
row2
[
3
+
bIdx
]
=
saturate_cast
<
uchar
>
((
y11
+
buv
)
>>
YUV420_SHIFT
);
}
}
if
(
j
%
4
==
2
)
{
u1
+=
stride
-
width
;
v1
+=
stride
-
width
;
}
}
}
}
}
};
};
...
@@ -2908,23 +2904,16 @@ struct YUV420p2RGBA8888Invoker
...
@@ -2908,23 +2904,16 @@ struct YUV420p2RGBA8888Invoker
Mat
*
dst
;
Mat
*
dst
;
const
uchar
*
my1
,
*
mu
,
*
mv
;
const
uchar
*
my1
,
*
mu
,
*
mv
;
int
width
,
stride
;
int
width
,
stride
;
int
ustepIdx
,
vstepIdx
;
YUV420p2RGBA8888Invoker
(
Mat
*
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
)
YUV420p2RGBA8888Invoker
(
Mat
*
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
,
int
_ustepIdx
,
int
_vstepIdx
)
:
dst
(
_dst
),
my1
(
_y1
),
mu
(
_u
),
mv
(
_v
),
width
(
_dst
->
cols
),
stride
(
_stride
)
{}
:
dst
(
_dst
),
my1
(
_y1
),
mu
(
_u
),
mv
(
_v
),
width
(
_dst
->
cols
),
stride
(
_stride
)
,
ustepIdx
(
_ustepIdx
),
vstepIdx
(
_vstepIdx
)
{}
void
operator
()(
const
BlockedRange
&
range
)
const
void
operator
()(
const
BlockedRange
&
range
)
const
{
{
int
rangeBegin
=
range
.
begin
()
*
2
;
int
rangeBegin
=
range
.
begin
()
*
2
;
int
rangeEnd
=
range
.
end
()
*
2
;
int
rangeEnd
=
range
.
end
()
*
2
;
//R = 1.164(Y - 16) + 1.596(V - 128)
//G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
//B = 1.164(Y - 16) + 2.018(U - 128)
//R = (1220542(Y - 16) + 1673527(V - 128) + (1 << 19)) >> 20
//G = (1220542(Y - 16) - 852492(V - 128) - 409993(U - 128) + (1 << 19)) >> 20
//B = (1220542(Y - 16) + 2116026(U - 128) + (1 << 19)) >> 20
const
int
cY
=
1220542
;
const
int
cY
=
1220542
;
const
int
cUB
=
2116026
;
const
int
cUB
=
2116026
;
const
int
cUG
=
-
409993
;
const
int
cUG
=
-
409993
;
...
@@ -2932,11 +2921,20 @@ struct YUV420p2RGBA8888Invoker
...
@@ -2932,11 +2921,20 @@ struct YUV420p2RGBA8888Invoker
const
int
cVR
=
1673527
;
const
int
cVR
=
1673527
;
const
int
YUV420_SHIFT
=
20
;
const
int
YUV420_SHIFT
=
20
;
size_t
uvsteps
[
2
]
=
{
width
/
2
,
stride
-
width
/
2
};
int
usIdx
=
ustepIdx
,
vsIdx
=
vstepIdx
;
const
uchar
*
y1
=
my1
+
rangeBegin
*
stride
;
const
uchar
*
y1
=
my1
+
rangeBegin
*
stride
;
const
uchar
*
u1
=
mu
+
(
range
.
begin
()
/
2
)
*
stride
+
(
range
.
begin
()
%
2
)
*
width
/
2
;
const
uchar
*
u1
=
mu
+
(
range
.
begin
()
/
2
)
*
stride
;
const
uchar
*
v1
=
mv
+
(
range
.
begin
()
/
2
)
*
stride
+
(
range
.
begin
()
%
2
)
*
width
/
2
;
const
uchar
*
v1
=
mv
+
(
range
.
begin
()
/
2
)
*
stride
;
for
(
int
j
=
rangeBegin
;
j
<
rangeEnd
;
j
+=
2
,
y1
+=
stride
*
2
,
u1
+=
width
/
2
,
v1
+=
width
/
2
)
if
(
range
.
begin
()
%
2
==
1
)
{
u1
+=
uvsteps
[(
usIdx
++
)
&
1
];
v1
+=
uvsteps
[(
vsIdx
++
)
&
1
];
}
for
(
int
j
=
rangeBegin
;
j
<
rangeEnd
;
j
+=
2
,
y1
+=
stride
*
2
,
u1
+=
uvsteps
[(
usIdx
++
)
&
1
],
v1
+=
uvsteps
[(
vsIdx
++
)
&
1
])
{
{
uchar
*
row1
=
dst
->
ptr
<
uchar
>
(
j
);
uchar
*
row1
=
dst
->
ptr
<
uchar
>
(
j
);
uchar
*
row2
=
dst
->
ptr
<
uchar
>
(
j
+
1
);
uchar
*
row2
=
dst
->
ptr
<
uchar
>
(
j
+
1
);
...
@@ -2975,12 +2973,6 @@ struct YUV420p2RGBA8888Invoker
...
@@ -2975,12 +2973,6 @@ struct YUV420p2RGBA8888Invoker
row2
[
4
+
bIdx
]
=
saturate_cast
<
uchar
>
((
y11
+
buv
)
>>
YUV420_SHIFT
);
row2
[
4
+
bIdx
]
=
saturate_cast
<
uchar
>
((
y11
+
buv
)
>>
YUV420_SHIFT
);
row2
[
7
]
=
uchar
(
0xff
);
row2
[
7
]
=
uchar
(
0xff
);
}
}
if
(
j
%
4
==
2
)
{
u1
+=
stride
-
width
;
v1
+=
stride
-
width
;
}
}
}
}
}
};
};
...
@@ -3012,9 +3004,9 @@ inline void cvtYUV420sp2RGBA(Mat& _dst, int _stride, const uchar* _y1, const uch
...
@@ -3012,9 +3004,9 @@ inline void cvtYUV420sp2RGBA(Mat& _dst, int _stride, const uchar* _y1, const uch
}
}
template
<
int
bIdx
>
template
<
int
bIdx
>
inline
void
cvtYUV420p2RGB
(
Mat
&
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
)
inline
void
cvtYUV420p2RGB
(
Mat
&
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
,
int
ustepIdx
,
int
vstepIdx
)
{
{
YUV420p2RGB888Invoker
<
bIdx
>
converter
(
&
_dst
,
_stride
,
_y1
,
_u
,
_v
);
YUV420p2RGB888Invoker
<
bIdx
>
converter
(
&
_dst
,
_stride
,
_y1
,
_u
,
_v
,
ustepIdx
,
vstepIdx
);
#ifdef HAVE_TBB
#ifdef HAVE_TBB
if
(
_dst
.
total
()
>=
MIN_SIZE_FOR_PARALLEL_YUV420_CONVERSION
)
if
(
_dst
.
total
()
>=
MIN_SIZE_FOR_PARALLEL_YUV420_CONVERSION
)
parallel_for
(
BlockedRange
(
0
,
_dst
.
rows
/
2
),
converter
);
parallel_for
(
BlockedRange
(
0
,
_dst
.
rows
/
2
),
converter
);
...
@@ -3024,9 +3016,9 @@ inline void cvtYUV420p2RGB(Mat& _dst, int _stride, const uchar* _y1, const uchar
...
@@ -3024,9 +3016,9 @@ inline void cvtYUV420p2RGB(Mat& _dst, int _stride, const uchar* _y1, const uchar
}
}
template
<
int
bIdx
>
template
<
int
bIdx
>
inline
void
cvtYUV420p2RGBA
(
Mat
&
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
)
inline
void
cvtYUV420p2RGBA
(
Mat
&
_dst
,
int
_stride
,
const
uchar
*
_y1
,
const
uchar
*
_u
,
const
uchar
*
_v
,
int
ustepIdx
,
int
vstepIdx
)
{
{
YUV420p2RGBA8888Invoker
<
bIdx
>
converter
(
&
_dst
,
_stride
,
_y1
,
_u
,
_v
);
YUV420p2RGBA8888Invoker
<
bIdx
>
converter
(
&
_dst
,
_stride
,
_y1
,
_u
,
_v
,
ustepIdx
,
vstepIdx
);
#ifdef HAVE_TBB
#ifdef HAVE_TBB
if
(
_dst
.
total
()
>=
MIN_SIZE_FOR_PARALLEL_YUV420_CONVERSION
)
if
(
_dst
.
total
()
>=
MIN_SIZE_FOR_PARALLEL_YUV420_CONVERSION
)
parallel_for
(
BlockedRange
(
0
,
_dst
.
rows
/
2
),
converter
);
parallel_for
(
BlockedRange
(
0
,
_dst
.
rows
/
2
),
converter
);
...
@@ -3461,15 +3453,19 @@ void cv::cvtColor( InputArray _src, OutputArray _dst, int code, int dcn )
...
@@ -3461,15 +3453,19 @@ void cv::cvtColor( InputArray _src, OutputArray _dst, int code, int dcn )
int
srcstep
=
(
int
)
src
.
step
;
int
srcstep
=
(
int
)
src
.
step
;
const
uchar
*
y
=
src
.
ptr
();
const
uchar
*
y
=
src
.
ptr
();
const
uchar
*
u
=
y
+
srcstep
*
dstSz
.
height
;
const
uchar
*
u
=
y
+
srcstep
*
dstSz
.
height
;
const
uchar
*
v
=
u
+
(
srcstep
*
dstSz
.
height
/
4
);
const
uchar
*
v
=
y
+
srcstep
*
(
dstSz
.
height
+
dstSz
.
height
/
4
)
+
(
dstSz
.
width
/
2
)
*
((
dstSz
.
height
%
4
)
/
2
);
if
(
uidx
==
1
)
std
::
swap
(
u
,
v
);
int
ustepIdx
=
0
;
int
vstepIdx
=
dstSz
.
height
%
4
==
2
?
1
:
0
;
if
(
uidx
==
1
)
{
std
::
swap
(
u
,
v
),
std
::
swap
(
ustepIdx
,
vstepIdx
);
};
switch
(
dcn
*
10
+
bidx
)
switch
(
dcn
*
10
+
bidx
)
{
{
case
30
:
cvtYUV420p2RGB
<
0
>
(
dst
,
srcstep
,
y
,
u
,
v
);
break
;
case
30
:
cvtYUV420p2RGB
<
0
>
(
dst
,
srcstep
,
y
,
u
,
v
,
ustepIdx
,
vstepIdx
);
break
;
case
32
:
cvtYUV420p2RGB
<
2
>
(
dst
,
srcstep
,
y
,
u
,
v
);
break
;
case
32
:
cvtYUV420p2RGB
<
2
>
(
dst
,
srcstep
,
y
,
u
,
v
,
ustepIdx
,
vstepIdx
);
break
;
case
40
:
cvtYUV420p2RGBA
<
0
>
(
dst
,
srcstep
,
y
,
u
,
v
);
break
;
case
40
:
cvtYUV420p2RGBA
<
0
>
(
dst
,
srcstep
,
y
,
u
,
v
,
ustepIdx
,
vstepIdx
);
break
;
case
42
:
cvtYUV420p2RGBA
<
2
>
(
dst
,
srcstep
,
y
,
u
,
v
);
break
;
case
42
:
cvtYUV420p2RGBA
<
2
>
(
dst
,
srcstep
,
y
,
u
,
v
,
ustepIdx
,
vstepIdx
);
break
;
default
:
CV_Error
(
CV_StsBadFlag
,
"Unknown/unsupported color conversion code"
);
break
;
default
:
CV_Error
(
CV_StsBadFlag
,
"Unknown/unsupported color conversion code"
);
break
;
};
};
}
}
...
...
modules/imgproc/test/test_cvtyuv.cpp
0 → 100644
View file @
470f6faf
#include "test_precomp.hpp"
using
namespace
cv
;
using
namespace
std
;
#undef RGB
#undef YUV
typedef
Vec3b
YUV
;
typedef
Vec3b
RGB
;
int
countOfDifferencies
(
const
Mat
&
gold
,
const
Mat
&
result
,
int
maxAllowedDifference
=
1
)
{
Mat
diff
;
absdiff
(
gold
,
result
,
diff
);
return
countNonZero
(
diff
.
reshape
(
1
)
>
maxAllowedDifference
);
}
class
YUVreader
{
public
:
virtual
~
YUVreader
()
{}
virtual
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
=
0
;
virtual
int
channels
()
=
0
;
virtual
Size
size
(
Size
imgSize
)
=
0
;
virtual
bool
requiresEvenHeight
()
{
return
true
;
}
virtual
bool
requiresEvenWidth
()
{
return
true
;
}
static
YUVreader
*
getReader
(
int
code
);
};
class
RGBwriter
{
public
:
virtual
~
RGBwriter
()
{}
virtual
void
write
(
Mat
&
rgb
,
int
row
,
int
col
,
const
RGB
&
val
)
=
0
;
virtual
int
channels
()
=
0
;
static
RGBwriter
*
getWriter
(
int
code
);
};
class
GRAYwriter
{
public
:
virtual
~
GRAYwriter
()
{}
virtual
void
write
(
Mat
&
gray
,
int
row
,
int
col
,
const
uchar
&
val
)
{
gray
.
at
<
uchar
>
(
row
,
col
)
=
val
;
}
virtual
int
channels
()
{
return
1
;
}
static
GRAYwriter
*
getWriter
(
int
code
);
};
class
RGB888Writer
:
public
RGBwriter
{
void
write
(
Mat
&
rgb
,
int
row
,
int
col
,
const
RGB
&
val
)
{
rgb
.
at
<
Vec3b
>
(
row
,
col
)
=
val
;
}
int
channels
()
{
return
3
;
}
};
class
BGR888Writer
:
public
RGBwriter
{
void
write
(
Mat
&
rgb
,
int
row
,
int
col
,
const
RGB
&
val
)
{
Vec3b
tmp
(
val
[
2
],
val
[
1
],
val
[
0
]);
rgb
.
at
<
Vec3b
>
(
row
,
col
)
=
tmp
;
}
int
channels
()
{
return
3
;
}
};
class
RGBA8888Writer
:
public
RGBwriter
{
void
write
(
Mat
&
rgb
,
int
row
,
int
col
,
const
RGB
&
val
)
{
Vec4b
tmp
(
val
[
0
],
val
[
1
],
val
[
2
],
255
);
rgb
.
at
<
Vec4b
>
(
row
,
col
)
=
tmp
;
}
int
channels
()
{
return
4
;
}
};
class
BGRA8888Writer
:
public
RGBwriter
{
void
write
(
Mat
&
rgb
,
int
row
,
int
col
,
const
RGB
&
val
)
{
Vec4b
tmp
(
val
[
2
],
val
[
1
],
val
[
0
],
255
);
rgb
.
at
<
Vec4b
>
(
row
,
col
)
=
tmp
;
}
int
channels
()
{
return
4
;
}
};
class
YUV420Reader
:
public
YUVreader
{
int
channels
()
{
return
1
;
}
Size
size
(
Size
imgSize
)
{
return
Size
(
imgSize
.
width
,
imgSize
.
height
*
3
/
2
);
}
};
class
YUV422Reader
:
public
YUVreader
{
int
channels
()
{
return
2
;
}
Size
size
(
Size
imgSize
)
{
return
imgSize
;
}
bool
requiresEvenHeight
()
{
return
false
;
}
};
class
NV21Reader
:
public
YUV420Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
uchar
y
=
yuv
.
ptr
<
uchar
>
(
row
)[
col
];
uchar
u
=
yuv
.
ptr
<
uchar
>
(
yuv
.
rows
*
2
/
3
+
row
/
2
)[(
col
/
2
)
*
2
+
1
];
uchar
v
=
yuv
.
ptr
<
uchar
>
(
yuv
.
rows
*
2
/
3
+
row
/
2
)[(
col
/
2
)
*
2
];
return
YUV
(
y
,
u
,
v
);
}
};
struct
NV12Reader
:
public
YUV420Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
uchar
y
=
yuv
.
ptr
<
uchar
>
(
row
)[
col
];
uchar
u
=
yuv
.
ptr
<
uchar
>
(
yuv
.
rows
*
2
/
3
+
row
/
2
)[(
col
/
2
)
*
2
];
uchar
v
=
yuv
.
ptr
<
uchar
>
(
yuv
.
rows
*
2
/
3
+
row
/
2
)[(
col
/
2
)
*
2
+
1
];
return
YUV
(
y
,
u
,
v
);
}
};
class
YV12Reader
:
public
YUV420Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
int
h
=
yuv
.
rows
*
2
/
3
;
uchar
y
=
yuv
.
ptr
<
uchar
>
(
row
)[
col
];
uchar
u
=
yuv
.
ptr
<
uchar
>
(
h
+
(
row
/
2
+
h
/
2
)
/
2
)[
col
/
2
+
((
row
/
2
+
h
/
2
)
%
2
)
*
(
yuv
.
cols
/
2
)];
uchar
v
=
yuv
.
ptr
<
uchar
>
(
h
+
row
/
4
)[
col
/
2
+
((
row
/
2
)
%
2
)
*
(
yuv
.
cols
/
2
)];
return
YUV
(
y
,
u
,
v
);
}
};
class
IYUVReader
:
public
YUV420Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
int
h
=
yuv
.
rows
*
2
/
3
;
uchar
y
=
yuv
.
ptr
<
uchar
>
(
row
)[
col
];
uchar
u
=
yuv
.
ptr
<
uchar
>
(
h
+
row
/
4
)[
col
/
2
+
((
row
/
2
)
%
2
)
*
(
yuv
.
cols
/
2
)];
uchar
v
=
yuv
.
ptr
<
uchar
>
(
h
+
(
row
/
2
+
h
/
2
)
/
2
)[
col
/
2
+
((
row
/
2
+
h
/
2
)
%
2
)
*
(
yuv
.
cols
/
2
)];
return
YUV
(
y
,
u
,
v
);
}
};
class
UYVYReader
:
public
YUV422Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
uchar
y
=
yuv
.
ptr
<
Vec2b
>
(
row
)[
col
][
1
];
uchar
u
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
][
0
];
uchar
v
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
+
1
][
0
];
return
YUV
(
y
,
u
,
v
);
}
};
class
YUY2Reader
:
public
YUV422Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
uchar
y
=
yuv
.
ptr
<
Vec2b
>
(
row
)[
col
][
0
];
uchar
u
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
][
1
];
uchar
v
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
+
1
][
1
];
return
YUV
(
y
,
u
,
v
);
}
};
class
YVYUReader
:
public
YUV422Reader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
uchar
y
=
yuv
.
ptr
<
Vec2b
>
(
row
)[
col
][
0
];
uchar
u
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
+
1
][
1
];
uchar
v
=
yuv
.
ptr
<
Vec2b
>
(
row
)[(
col
/
2
)
*
2
][
1
];
return
YUV
(
y
,
u
,
v
);
}
};
class
YUV888Reader
:
public
YUVreader
{
YUV
read
(
const
Mat
&
yuv
,
int
row
,
int
col
)
{
return
yuv
.
at
<
YUV
>
(
row
,
col
);
}
int
channels
()
{
return
3
;
}
Size
size
(
Size
imgSize
)
{
return
imgSize
;
}
bool
requiresEvenHeight
()
{
return
false
;
}
bool
requiresEvenWidth
()
{
return
false
;
}
};
class
YUV2RGB_Converter
{
public
:
RGB
convert
(
YUV
yuv
)
{
int
y
=
std
::
max
(
0
,
yuv
[
0
]
-
16
);
int
u
=
yuv
[
1
]
-
128
;
int
v
=
yuv
[
2
]
-
128
;
uchar
r
=
saturate_cast
<
uchar
>
(
1.164
f
*
y
+
1.596
f
*
v
);
uchar
g
=
saturate_cast
<
uchar
>
(
1.164
f
*
y
-
0.813
f
*
v
-
0.391
f
*
u
);
uchar
b
=
saturate_cast
<
uchar
>
(
1.164
f
*
y
+
2.018
f
*
u
);
return
RGB
(
r
,
g
,
b
);
}
};
class
YUV2GRAY_Converter
{
public
:
uchar
convert
(
YUV
yuv
)
{
return
yuv
[
0
];
}
};
YUVreader
*
YUVreader
::
getReader
(
int
code
)
{
switch
(
code
)
{
case
CV_YUV2RGB_NV12
:
case
CV_YUV2BGR_NV12
:
case
CV_YUV2RGBA_NV12
:
case
CV_YUV2BGRA_NV12
:
return
new
NV12Reader
();
case
CV_YUV2RGB_NV21
:
case
CV_YUV2BGR_NV21
:
case
CV_YUV2RGBA_NV21
:
case
CV_YUV2BGRA_NV21
:
return
new
NV21Reader
();
case
CV_YUV2RGB_YV12
:
case
CV_YUV2BGR_YV12
:
case
CV_YUV2RGBA_YV12
:
case
CV_YUV2BGRA_YV12
:
return
new
YV12Reader
();
case
CV_YUV2RGB_IYUV
:
case
CV_YUV2BGR_IYUV
:
case
CV_YUV2RGBA_IYUV
:
case
CV_YUV2BGRA_IYUV
:
return
new
IYUVReader
();
case
CV_YUV2RGB_UYVY
:
case
CV_YUV2BGR_UYVY
:
case
CV_YUV2RGBA_UYVY
:
case
CV_YUV2BGRA_UYVY
:
return
new
UYVYReader
();
//case CV_YUV2RGB_VYUY = 109,
//case CV_YUV2BGR_VYUY = 110,
//case CV_YUV2RGBA_VYUY = 113,
//case CV_YUV2BGRA_VYUY = 114,
// return ??
case
CV_YUV2RGB_YUY2
:
case
CV_YUV2BGR_YUY2
:
case
CV_YUV2RGBA_YUY2
:
case
CV_YUV2BGRA_YUY2
:
return
new
YUY2Reader
();
case
CV_YUV2RGB_YVYU
:
case
CV_YUV2BGR_YVYU
:
case
CV_YUV2RGBA_YVYU
:
case
CV_YUV2BGRA_YVYU
:
return
new
YVYUReader
();
case
CV_YUV2GRAY_420
:
return
new
NV21Reader
();
case
CV_YUV2GRAY_UYVY
:
return
new
UYVYReader
();
case
CV_YUV2GRAY_YUY2
:
return
new
YUY2Reader
();
case
CV_YUV2BGR
:
case
CV_YUV2RGB
:
return
new
YUV888Reader
();
default
:
return
0
;
}
}
RGBwriter
*
RGBwriter
::
getWriter
(
int
code
)
{
switch
(
code
)
{
case
CV_YUV2RGB_NV12
:
case
CV_YUV2RGB_NV21
:
case
CV_YUV2RGB_YV12
:
case
CV_YUV2RGB_IYUV
:
case
CV_YUV2RGB_UYVY
:
//case CV_YUV2RGB_VYUY:
case
CV_YUV2RGB_YUY2
:
case
CV_YUV2RGB_YVYU
:
case
CV_YUV2RGB
:
return
new
RGB888Writer
();
case
CV_YUV2BGR_NV12
:
case
CV_YUV2BGR_NV21
:
case
CV_YUV2BGR_YV12
:
case
CV_YUV2BGR_IYUV
:
case
CV_YUV2BGR_UYVY
:
//case CV_YUV2BGR_VYUY:
case
CV_YUV2BGR_YUY2
:
case
CV_YUV2BGR_YVYU
:
case
CV_YUV2BGR
:
return
new
BGR888Writer
();
case
CV_YUV2RGBA_NV12
:
case
CV_YUV2RGBA_NV21
:
case
CV_YUV2RGBA_YV12
:
case
CV_YUV2RGBA_IYUV
:
case
CV_YUV2RGBA_UYVY
:
//case CV_YUV2RGBA_VYUY:
case
CV_YUV2RGBA_YUY2
:
case
CV_YUV2RGBA_YVYU
:
return
new
RGBA8888Writer
();
case
CV_YUV2BGRA_NV12
:
case
CV_YUV2BGRA_NV21
:
case
CV_YUV2BGRA_YV12
:
case
CV_YUV2BGRA_IYUV
:
case
CV_YUV2BGRA_UYVY
:
//case CV_YUV2BGRA_VYUY:
case
CV_YUV2BGRA_YUY2
:
case
CV_YUV2BGRA_YVYU
:
return
new
BGRA8888Writer
();
default
:
return
0
;
};
}
GRAYwriter
*
GRAYwriter
::
getWriter
(
int
code
)
{
switch
(
code
)
{
case
CV_YUV2GRAY_420
:
case
CV_YUV2GRAY_UYVY
:
case
CV_YUV2GRAY_YUY2
:
return
new
GRAYwriter
();
default
:
return
0
;
}
}
template
<
class
convertor
>
void
referenceYUV2RGB
(
const
Mat
&
yuv
,
Mat
&
rgb
,
YUVreader
*
yuvReader
,
RGBwriter
*
rgbWriter
)
{
convertor
cvt
;
for
(
int
row
=
0
;
row
<
rgb
.
rows
;
++
row
)
for
(
int
col
=
0
;
col
<
rgb
.
cols
;
++
col
)
rgbWriter
->
write
(
rgb
,
row
,
col
,
cvt
.
convert
(
yuvReader
->
read
(
yuv
,
row
,
col
)));
}
template
<
class
convertor
>
void
referenceYUV2GRAY
(
const
Mat
&
yuv
,
Mat
&
rgb
,
YUVreader
*
yuvReader
,
GRAYwriter
*
grayWriter
)
{
convertor
cvt
;
for
(
int
row
=
0
;
row
<
rgb
.
rows
;
++
row
)
for
(
int
col
=
0
;
col
<
rgb
.
cols
;
++
col
)
grayWriter
->
write
(
rgb
,
row
,
col
,
cvt
.
convert
(
yuvReader
->
read
(
yuv
,
row
,
col
)));
}
CV_ENUM
(
YUVCVTS
,
CV_YUV2RGB_NV12
,
CV_YUV2BGR_NV12
,
CV_YUV2RGB_NV21
,
CV_YUV2BGR_NV21
,
CV_YUV2RGBA_NV12
,
CV_YUV2BGRA_NV12
,
CV_YUV2RGBA_NV21
,
CV_YUV2BGRA_NV21
,
CV_YUV2RGB_YV12
,
CV_YUV2BGR_YV12
,
CV_YUV2RGB_IYUV
,
CV_YUV2BGR_IYUV
,
CV_YUV2RGBA_YV12
,
CV_YUV2BGRA_YV12
,
CV_YUV2RGBA_IYUV
,
CV_YUV2BGRA_IYUV
,
CV_YUV2RGB_UYVY
,
CV_YUV2BGR_UYVY
,
CV_YUV2RGBA_UYVY
,
CV_YUV2BGRA_UYVY
,
CV_YUV2RGB_YUY2
,
CV_YUV2BGR_YUY2
,
CV_YUV2RGB_YVYU
,
CV_YUV2BGR_YVYU
,
CV_YUV2RGBA_YUY2
,
CV_YUV2BGRA_YUY2
,
CV_YUV2RGBA_YVYU
,
CV_YUV2BGRA_YVYU
,
CV_YUV2GRAY_420
,
CV_YUV2GRAY_UYVY
,
CV_YUV2GRAY_YUY2
,
CV_YUV2BGR
,
CV_YUV2RGB
);
typedef
::
testing
::
TestWithParam
<
YUVCVTS
>
Imgproc_ColorYUV
;
TEST_P
(
Imgproc_ColorYUV
,
accuracy
)
{
int
code
=
GetParam
();
RNG
&
random
=
theRNG
();
YUVreader
*
yuvReader
=
YUVreader
::
getReader
(
code
);
RGBwriter
*
rgbWriter
=
RGBwriter
::
getWriter
(
code
);
GRAYwriter
*
grayWriter
=
GRAYwriter
::
getWriter
(
code
);
int
dcn
=
(
rgbWriter
==
0
)
?
grayWriter
->
channels
()
:
rgbWriter
->
channels
();
for
(
int
iter
=
0
;
iter
<
30
;
++
iter
)
{
Size
sz
(
random
.
uniform
(
1
,
641
),
random
.
uniform
(
1
,
481
));
if
(
yuvReader
->
requiresEvenWidth
())
sz
.
width
+=
sz
.
width
%
2
;
if
(
yuvReader
->
requiresEvenHeight
())
sz
.
height
+=
sz
.
height
%
2
;
Size
ysz
=
yuvReader
->
size
(
sz
);
Mat
src
=
Mat
(
ysz
.
height
,
ysz
.
width
*
yuvReader
->
channels
(),
CV_8UC1
).
reshape
(
yuvReader
->
channels
());
Mat
dst
=
Mat
(
sz
.
height
,
sz
.
width
*
dcn
,
CV_8UC1
).
reshape
(
dcn
);
Mat
gold
(
sz
,
CV_8UC
(
dcn
));
random
.
fill
(
src
,
RNG
::
UNIFORM
,
0
,
256
);
if
(
rgbWriter
)
referenceYUV2RGB
<
YUV2RGB_Converter
>
(
src
,
gold
,
yuvReader
,
rgbWriter
);
else
referenceYUV2GRAY
<
YUV2GRAY_Converter
>
(
src
,
gold
,
yuvReader
,
grayWriter
);
cv
::
cvtColor
(
src
,
dst
,
code
,
-
1
);
EXPECT_EQ
(
0
,
countOfDifferencies
(
gold
,
dst
));
}
}
TEST_P
(
Imgproc_ColorYUV
,
roi_accuracy
)
{
int
code
=
GetParam
();
RNG
&
random
=
theRNG
();
YUVreader
*
yuvReader
=
YUVreader
::
getReader
(
code
);
RGBwriter
*
rgbWriter
=
RGBwriter
::
getWriter
(
code
);
GRAYwriter
*
grayWriter
=
GRAYwriter
::
getWriter
(
code
);
int
dcn
=
(
rgbWriter
==
0
)
?
grayWriter
->
channels
()
:
rgbWriter
->
channels
();
for
(
int
iter
=
0
;
iter
<
30
;
++
iter
)
{
Size
sz
(
random
.
uniform
(
1
,
641
),
random
.
uniform
(
1
,
481
));
if
(
yuvReader
->
requiresEvenWidth
())
sz
.
width
+=
sz
.
width
%
2
;
if
(
yuvReader
->
requiresEvenHeight
())
sz
.
height
+=
sz
.
height
%
2
;
int
roi_offset_top
=
random
.
uniform
(
0
,
6
);
int
roi_offset_bottom
=
random
.
uniform
(
0
,
6
);
int
roi_offset_left
=
random
.
uniform
(
0
,
6
);
int
roi_offset_right
=
random
.
uniform
(
0
,
6
);
Size
ysz
=
yuvReader
->
size
(
sz
);
Mat
src_full
(
ysz
.
height
+
roi_offset_top
+
roi_offset_bottom
,
ysz
.
width
+
roi_offset_left
+
roi_offset_right
,
CV_8UC
(
yuvReader
->
channels
()));
Mat
dst_full
(
sz
.
height
+
roi_offset_left
+
roi_offset_right
,
sz
.
width
+
roi_offset_top
+
roi_offset_bottom
,
CV_8UC
(
dcn
),
Scalar
::
all
(
0
));
Mat
gold_full
(
dst_full
.
size
(),
CV_8UC
(
dcn
),
Scalar
::
all
(
0
));
random
.
fill
(
src_full
,
RNG
::
UNIFORM
,
0
,
256
);
Mat
src
=
src_full
(
Range
(
roi_offset_top
,
roi_offset_top
+
ysz
.
height
),
Range
(
roi_offset_left
,
roi_offset_left
+
ysz
.
width
));
Mat
dst
=
dst_full
(
Range
(
roi_offset_left
,
roi_offset_left
+
sz
.
height
),
Range
(
roi_offset_top
,
roi_offset_top
+
sz
.
width
));
Mat
gold
=
gold_full
(
Range
(
roi_offset_left
,
roi_offset_left
+
sz
.
height
),
Range
(
roi_offset_top
,
roi_offset_top
+
sz
.
width
));
if
(
rgbWriter
)
referenceYUV2RGB
<
YUV2RGB_Converter
>
(
src
,
gold
,
yuvReader
,
rgbWriter
);
else
referenceYUV2GRAY
<
YUV2GRAY_Converter
>
(
src
,
gold
,
yuvReader
,
grayWriter
);
cv
::
cvtColor
(
src
,
dst
,
code
,
-
1
);
EXPECT_EQ
(
0
,
countOfDifferencies
(
gold_full
,
dst_full
));
}
}
INSTANTIATE_TEST_CASE_P
(
cvt420
,
Imgproc_ColorYUV
,
::
testing
::
Values
(
CV_YUV2RGB_NV12
,
CV_YUV2BGR_NV12
,
CV_YUV2RGB_NV21
,
CV_YUV2BGR_NV21
,
CV_YUV2RGBA_NV12
,
CV_YUV2BGRA_NV12
,
CV_YUV2RGBA_NV21
,
CV_YUV2BGRA_NV21
,
CV_YUV2RGB_YV12
,
CV_YUV2BGR_YV12
,
CV_YUV2RGB_IYUV
,
CV_YUV2BGR_IYUV
,
CV_YUV2RGBA_YV12
,
CV_YUV2BGRA_YV12
,
CV_YUV2RGBA_IYUV
,
CV_YUV2BGRA_IYUV
,
CV_YUV2GRAY_420
));
INSTANTIATE_TEST_CASE_P
(
DISABLED_cvt888
,
Imgproc_ColorYUV
,
::
testing
::
Values
(
CV_YUV2BGR
,
CV_YUV2RGB
));
INSTANTIATE_TEST_CASE_P
(
DISABLED_cvt422
,
Imgproc_ColorYUV
,
::
testing
::
Values
(
CV_YUV2RGB_UYVY
,
CV_YUV2BGR_UYVY
,
CV_YUV2RGBA_UYVY
,
CV_YUV2BGRA_UYVY
,
CV_YUV2RGB_YUY2
,
CV_YUV2BGR_YUY2
,
CV_YUV2RGB_YVYU
,
CV_YUV2BGR_YVYU
,
CV_YUV2RGBA_YUY2
,
CV_YUV2BGRA_YUY2
,
CV_YUV2RGBA_YVYU
,
CV_YUV2BGRA_YVYU
,
CV_YUV2GRAY_UYVY
,
CV_YUV2GRAY_YUY2
));
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment