48 lines
1.6 KiB
C++
48 lines
1.6 KiB
C++
#include <iostream>
|
||
#include <string>
|
||
// 包含SpireCV SDK头文件
|
||
#include <sv_world.h>
|
||
|
||
using namespace std;
|
||
|
||
int main(int argc, char *argv[]) {
|
||
// 实例化
|
||
sv::CommonObjectDetector cod;
|
||
// 手动导入相机参数，如果使用Amov的G1等吊舱或相机，则可以忽略该步骤，将自动下载相机参数文件
|
||
cod.loadCameraParams(sv::get_home() + "/SpireCV/confs/calib_webcam_1280x720.yaml");
|
||
cod.loadAlgorithmParams(sv::get_home() + "/SpireCV/confs/sv_algorithm_params.json");
|
||
sv::MultipleObjectTracker mot;
|
||
// 手动导入相机参数，如果使用Amov的G1等吊舱或相机，则可以忽略该步骤，将自动下载相机参数文件
|
||
mot.loadCameraParams(sv::get_home() + "/SpireCV/confs/calib_webcam_1280x720.yaml");
|
||
mot.loadAlgorithmParams(sv::get_home() + "/SpireCV/confs/sv_algorithm_params.json");
|
||
mot.init(&cod);
|
||
|
||
// 打开摄像头
|
||
sv::Camera cap;
|
||
cap.setWH(mot.image_width, mot.image_height);
|
||
cap.setFps(30);
|
||
cap.open(sv::CameraType::V4L2CAM, 0); // CameraID 0
|
||
// 实例化OpenCV的Mat类，用于内存单帧图像
|
||
cv::Mat img;
|
||
int frame_id = 0;
|
||
while (1)
|
||
{
|
||
// 实例化SpireCV的 单帧检测结果 接口类 TargetsInFrame
|
||
sv::TargetsInFrame tgts(frame_id++);
|
||
// 读取一帧图像到img
|
||
cap.read(img);
|
||
cv::resize(img, img, cv::Size(mot.image_width, mot.image_height));
|
||
|
||
// 执行通用目标检测
|
||
sv::TargetsInFrame person_tgts = mot.track(img, tgts);
|
||
// 可视化检测结果，叠加到img上
|
||
sv::drawTargetsInFrame(img, person_tgts);
|
||
|
||
// 显示检测结果img
|
||
cv::imshow("img", img);
|
||
cv::waitKey(10);
|
||
}
|
||
|
||
return 0;
|
||
}
|