Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Greenplum
Opencv
提交
42459cad
O
Opencv
项目概览
Greenplum
/
Opencv
11 个月 前同步成功
通知
7
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
Opencv
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
42459cad
编写于
1月 18, 2018
作者:
V
Vadim Pisarevsky
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #10634 from sturkmen72:update_samples
上级
0def2dbb
4046d9f8
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
29 addition
and
33 deletion
+29
-33
samples/dnn/resnet_ssd_face.cpp
samples/dnn/resnet_ssd_face.cpp
+3
-5
samples/dnn/ssd_mobilenet_object_detection.cpp
samples/dnn/ssd_mobilenet_object_detection.cpp
+18
-16
samples/dnn/ssd_object_detection.cpp
samples/dnn/ssd_object_detection.cpp
+4
-6
samples/dnn/yolo_object_detection.cpp
samples/dnn/yolo_object_detection.cpp
+4
-6
未找到文件。
samples/dnn/resnet_ssd_face.cpp
浏览文件 @
42459cad
#include <opencv2/dnn.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <iostream>
using
namespace
cv
;
using
namespace
cv
::
dnn
;
#include <iostream>
#include <cstdlib>
using
namespace
std
;
using
namespace
cv
::
dnn
;
const
size_t
inWidth
=
300
;
const
size_t
inHeight
=
300
;
...
...
@@ -152,7 +150,7 @@ int main(int argc, char** argv)
Size
labelSize
=
getTextSize
(
label
,
FONT_HERSHEY_SIMPLEX
,
0.5
,
1
,
&
baseLine
);
rectangle
(
frame
,
Rect
(
Point
(
xLeftBottom
,
yLeftBottom
-
labelSize
.
height
),
Size
(
labelSize
.
width
,
labelSize
.
height
+
baseLine
)),
Scalar
(
255
,
255
,
255
),
CV_
FILLED
);
Scalar
(
255
,
255
,
255
),
FILLED
);
putText
(
frame
,
label
,
Point
(
xLeftBottom
,
yLeftBottom
),
FONT_HERSHEY_SIMPLEX
,
0.5
,
Scalar
(
0
,
0
,
0
));
}
...
...
samples/dnn/ssd_mobilenet_object_detection.cpp
浏览文件 @
42459cad
...
...
@@ -2,14 +2,11 @@
#include <opencv2/dnn/shape_utils.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <iostream>
using
namespace
cv
;
using
namespace
cv
::
dnn
;
#include <fstream>
#include <iostream>
#include <cstdlib>
using
namespace
std
;
using
namespace
cv
::
dnn
;
const
size_t
inWidth
=
300
;
const
size_t
inHeight
=
300
;
...
...
@@ -22,11 +19,13 @@ const char* classNames[] = {"background",
"motorbike"
,
"person"
,
"pottedplant"
,
"sheep"
,
"sofa"
,
"train"
,
"tvmonitor"
};
const
char
*
param
s
const
String
key
s
=
"{ help | false | print usage }"
"{ proto | MobileNetSSD_deploy.prototxt | model configuration }"
"{ proto | MobileNetSSD_deploy.prototxt
| model configuration }"
"{ model | MobileNetSSD_deploy.caffemodel | model weights }"
"{ camera_device | 0 | camera device number }"
"{ camera_width | 640 | camera device width }"
"{ camera_height | 480 | camera device height }"
"{ video | | video or image for detection}"
"{ out | | path to output video file}"
"{ min_confidence | 0.2 | min confidence }"
...
...
@@ -35,7 +34,7 @@ const char* params
int
main
(
int
argc
,
char
**
argv
)
{
CommandLineParser
parser
(
argc
,
argv
,
param
s
);
CommandLineParser
parser
(
argc
,
argv
,
key
s
);
parser
.
about
(
"This sample uses MobileNet Single-Shot Detector "
"(https://arxiv.org/abs/1704.04861) "
"to detect objects on camera/video/image.
\n
"
...
...
@@ -43,14 +42,14 @@ int main(int argc, char** argv)
"https://github.com/chuanqi305/MobileNet-SSD
\n
"
"Default network is 300x300 and 20-classes VOC.
\n
"
);
if
(
parser
.
get
<
bool
>
(
"help"
)
||
argc
==
1
)
if
(
parser
.
get
<
bool
>
(
"help"
))
{
parser
.
printMessage
();
return
0
;
}
String
modelConfiguration
=
parser
.
get
<
s
tring
>
(
"proto"
);
String
modelBinary
=
parser
.
get
<
s
tring
>
(
"model"
);
String
modelConfiguration
=
parser
.
get
<
S
tring
>
(
"proto"
);
String
modelBinary
=
parser
.
get
<
S
tring
>
(
"model"
);
CV_Assert
(
!
modelConfiguration
.
empty
()
&&
!
modelBinary
.
empty
());
//! [Initialize network]
...
...
@@ -82,6 +81,9 @@ int main(int argc, char** argv)
cout
<<
"Couldn't find camera: "
<<
cameraDevice
<<
endl
;
return
-
1
;
}
cap
.
set
(
CAP_PROP_FRAME_WIDTH
,
parser
.
get
<
int
>
(
"camera_width"
));
cap
.
set
(
CAP_PROP_FRAME_HEIGHT
,
parser
.
get
<
int
>
(
"camera_height"
));
}
else
{
...
...
@@ -94,11 +96,11 @@ int main(int argc, char** argv)
}
//Acquire input size
Size
inVideoSize
((
int
)
cap
.
get
(
C
V_C
AP_PROP_FRAME_WIDTH
),
(
int
)
cap
.
get
(
C
V_C
AP_PROP_FRAME_HEIGHT
));
Size
inVideoSize
((
int
)
cap
.
get
(
CAP_PROP_FRAME_WIDTH
),
(
int
)
cap
.
get
(
CAP_PROP_FRAME_HEIGHT
));
double
fps
=
cap
.
get
(
C
V_C
AP_PROP_FPS
);
int
fourcc
=
static_cast
<
int
>
(
cap
.
get
(
C
V_C
AP_PROP_FOURCC
));
double
fps
=
cap
.
get
(
CAP_PROP_FPS
);
int
fourcc
=
static_cast
<
int
>
(
cap
.
get
(
CAP_PROP_FOURCC
));
VideoWriter
outputVideo
;
outputVideo
.
open
(
parser
.
get
<
String
>
(
"out"
)
,
(
fourcc
!=
0
?
fourcc
:
VideoWriter
::
fourcc
(
'M'
,
'J'
,
'P'
,
'G'
)),
...
...
@@ -168,7 +170,7 @@ int main(int argc, char** argv)
top
=
max
(
top
,
labelSize
.
height
);
rectangle
(
frame
,
Point
(
left
,
top
-
labelSize
.
height
),
Point
(
left
+
labelSize
.
width
,
top
+
baseLine
),
Scalar
(
255
,
255
,
255
),
CV_
FILLED
);
Scalar
(
255
,
255
,
255
),
FILLED
);
putText
(
frame
,
label
,
Point
(
left
,
top
),
FONT_HERSHEY_SIMPLEX
,
0.5
,
Scalar
(
0
,
0
,
0
));
}
...
...
samples/dnn/ssd_object_detection.cpp
浏览文件 @
42459cad
...
...
@@ -2,13 +2,11 @@
#include <opencv2/dnn/shape_utils.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
using
namespace
cv
;
using
namespace
cv
::
dnn
;
#include <fstream>
#include <iostream>
#include <cstdlib>
using
namespace
cv
;
using
namespace
std
;
using
namespace
cv
::
dnn
;
const
char
*
classNames
[]
=
{
"background"
,
"aeroplane"
,
"bicycle"
,
"bird"
,
"boat"
,
...
...
@@ -144,7 +142,7 @@ int main(int argc, char** argv)
Size
labelSize
=
getTextSize
(
label
,
FONT_HERSHEY_SIMPLEX
,
0.5
,
1
,
&
baseLine
);
rectangle
(
frame
,
Rect
(
Point
(
xLeftBottom
,
yLeftBottom
-
labelSize
.
height
),
Size
(
labelSize
.
width
,
labelSize
.
height
+
baseLine
)),
Scalar
(
255
,
255
,
255
),
CV_
FILLED
);
Scalar
(
255
,
255
,
255
),
FILLED
);
putText
(
frame
,
label
,
Point
(
xLeftBottom
,
yLeftBottom
),
FONT_HERSHEY_SIMPLEX
,
0.5
,
Scalar
(
0
,
0
,
0
));
}
...
...
samples/dnn/yolo_object_detection.cpp
浏览文件 @
42459cad
...
...
@@ -7,8 +7,6 @@
#include <opencv2/highgui.hpp>
#include <fstream>
#include <iostream>
#include <algorithm>
#include <cstdlib>
using
namespace
std
;
using
namespace
cv
;
...
...
@@ -26,7 +24,7 @@ static const char* params =
"{ model | | model weights }"
"{ camera_device | 0 | camera device number}"
"{ source | | video or image for detection}"
"{
save | | file name output
}"
"{
out | | path to output video file
}"
"{ fps | 3 | frame per second }"
"{ style | box | box or line style draw }"
"{ min_confidence | 0.24 | min confidence }"
...
...
@@ -84,9 +82,9 @@ int main(int argc, char** argv)
}
}
if
(
!
parser
.
get
<
String
>
(
"
save
"
).
empty
())
if
(
!
parser
.
get
<
String
>
(
"
out
"
).
empty
())
{
writer
.
open
(
parser
.
get
<
String
>
(
"
save
"
),
codec
,
fps
,
Size
((
int
)
cap
.
get
(
CAP_PROP_FRAME_WIDTH
),(
int
)
cap
.
get
(
CAP_PROP_FRAME_HEIGHT
)),
1
);
writer
.
open
(
parser
.
get
<
String
>
(
"
out
"
),
codec
,
fps
,
Size
((
int
)
cap
.
get
(
CAP_PROP_FRAME_WIDTH
),(
int
)
cap
.
get
(
CAP_PROP_FRAME_HEIGHT
)),
1
);
}
vector
<
String
>
classNamesVec
;
...
...
@@ -169,7 +167,7 @@ int main(int argc, char** argv)
int
baseLine
=
0
;
Size
labelSize
=
getTextSize
(
label
,
FONT_HERSHEY_SIMPLEX
,
0.5
,
1
,
&
baseLine
);
rectangle
(
frame
,
Rect
(
p1
,
Size
(
labelSize
.
width
,
labelSize
.
height
+
baseLine
)),
object_roi_color
,
CV_
FILLED
);
object_roi_color
,
FILLED
);
putText
(
frame
,
label
,
p1
+
Point
(
0
,
labelSize
.
height
),
FONT_HERSHEY_SIMPLEX
,
0.5
,
Scalar
(
0
,
0
,
0
));
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录