各位好,请问这个C++的代码怎么写成python语言?

C罗子 2017-12-25 01:26:39
#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/tracking/tracker.hpp>
#include <boost/filesystem.hpp>

const cv::Size MAX_DETECT_SIZE = cv::Size(100, 200);
const int MAX_MISS_FRAME = 10;
const double MIN_NEW_DETECT_INTERSECTION_RATE = 0.5;

class MyTracker {
private:
static int next_id;
int id;
int n_miss_frame = 0;
cv::Rect2d rect;
cv::Ptr<cv::Tracker> cv_tracker;
public:

MyTracker(const cv::Mat& _frame, const cv::Rect2d& _rect)
: id(next_id++), rect(_rect) {
cv_tracker = cv::Tracker::create("BOOSTING"); // or "MIL"
cv_tracker->init(_frame, _rect);
}

bool update(const cv::Mat& _frame){
n_miss_frame++;
return cv_tracker->update(_frame, rect) && n_miss_frame < MAX_MISS_FRAME;
}

bool registerNewDetect(const cv::Rect2d& _new_detect){
double intersection_rate = 1.0 * (_new_detect & rect).area() / (_new_detect | rect).area();
bool is_registered = intersection_rate > MIN_NEW_DETECT_INTERSECTION_RATE;
if (is_registered) n_miss_frame = 0;
return is_registered;
}

void draw(cv::Mat& _image) const{
cv::rectangle(_image, rect, cv::Scalar(255, 0, 0), 2, 1);
cv::putText(_image, cv::format("%03d", id), cv::Point(rect.x + 5, rect.y + 17),
cv::FONT_HERSHEY_SIMPLEX, 0.5, cv::Scalar(255,0,0), 1, CV_AA);
}
};
int MyTracker::next_id = 0;


int main(int argc, char* argv[]){
if(argc != 2){
std::cout << "usage: " << argv[0] << " videodir" << std::endl;
exit(1);
}

namespace fs = boost::filesystem;
std::vector<std::string> frame_paths;
for(auto it = fs::directory_iterator(argv[1]); it != fs::directory_iterator(); ++it){
frame_paths.push_back(it->path().string());
}

cv::HOGDescriptor detector;
detector.setSVMDetector(cv::HOGDescriptor::getDefaultPeopleDetector());
std::vector<MyTracker> trackers;

for (auto& frame_path : frame_paths){
std::cout << "frame : " << frame_path << std::endl;
cv::Mat frame = cv::imread(frame_path);
std::vector<cv::Rect> detections;
detector.detectMultiScale(frame, detections);
for (auto t_it = trackers.begin(); t_it != trackers.end();){
t_it = (t_it->update(frame)) ? std::next(t_it) : trackers.erase(t_it);
}

for(auto& d_rect : detections){
if (d_rect.size().area() > MAX_DETECT_SIZE.area()) continue;
bool exists = std::any_of(trackers.begin(), trackers.end(),
[&d_rect](MyTracker& t){return t.registerNewDetect(d_rect);});
if(!exists) trackers.push_back(MyTracker(frame, d_rect));
}

cv::Mat image = frame.clone();
for(auto& t : trackers) t.draw(image);
for(auto& d_rect : detections) cv::rectangle(image, d_rect, cv::Scalar(0, 255, 0), 2, 1);
cv::imshow("demo", image);
cv::waitKey(1);
}
return 0;
}


我现在需要把这个代码写成python语言,但是我又不懂C++,请各位前辈帮忙。
...全文
154 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
oyljerry 2017-12-25
  • 打赏
  • 举报
回复
python有opencv的库,然后就需要你自己对应的调用

37,719

社区成员

发帖
与我相关
我的任务
社区描述
JavaScript,VBScript,AngleScript,ActionScript,Shell,Perl,Ruby,Lua,Tcl,Scala,MaxScript 等脚本语言交流。
社区管理员
  • 脚本语言(Perl/Python)社区
  • IT.BOB
加入社区
  • 近7日
  • 近30日
  • 至今

试试用AI创作助手写篇文章吧