Opencv 3.4 目标跟踪 goturn算法 实现

http://blog.csdn.net/dark5669/article/details/78579110

这里是一篇问题汇总。

对于VS2017在按照相关配置,把opencv的和opencontrib源码下载编译之后,配置好opencv后(配置请参考上一篇文章)

对于python 在anaconda上的有opencv3.3.1的contrib编译好的,https://www.lfd.uci.edu/~gohlke/pythonlibs/到这里下载就行注意版本

如果是opencv3.4,需要重新编译python版本的,试了一下失败了。

下面是算法实现。

Python版本

import cv2

import sys

(major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.')

if __name__ =='__main__' :

# Set up tracker.

# Instead of MIL, you can also use

    tracker_types = ['BOOSTING','MIL','KCF','TLD','MEDIANFLOW','GOTURN']

tracker_type = tracker_types[4]

if int(minor_ver) <3:

tracker = cv2.Tracker_create(tracker_type)

else:

if tracker_type =='BOOSTING':

tracker = cv2.TrackerBoosting_create()

if tracker_type =='MIL':

tracker = cv2.TrackerMIL_create()

if tracker_type =='KCF':

tracker = cv2.TrackerKCF_create()

if tracker_type =='TLD':

tracker = cv2.TrackerTLD_create()

if tracker_type =='MEDIANFLOW':

tracker = cv2.TrackerMedianFlow_create()

if tracker_type =='GOTURN':

tracker = cv2.TrackerGOTURN_create()

# Read video

#    video = cv2.VideoCapture("videos/chaplin.mp4")

    video = cv2.VideoCapture(0)

# Exit if video not opened.

    if not video.isOpened():

print ("Could not open video")

sys.exit()

# Read first frame.

    ok, frame = video.read()

if not ok:

print ('Cannot read video file')

sys.exit()

# Define an initial bounding box

    bbox = (287,23,86,320)

# Uncomment the line below to select a different bounding box

    bbox = cv2.selectROI(frame,False)

# Initialize tracker with first frame and bounding box

    ok = tracker.init(frame, bbox)

while True:

# Read a new frame

        ok, frame = video.read()

if not ok:

break

        # Start timer

        timer = cv2.getTickCount()

# Update tracker

        ok, bbox = tracker.update(frame)

# Calculate Frames per second (FPS)

        fps = cv2.getTickFrequency() / (cv2.getTickCount() - timer);

# Draw bounding box

        if ok:

# Tracking success

            p1 = (int(bbox[0]),int(bbox[1]))

p2 = (int(bbox[0] + bbox[2]),int(bbox[1] + bbox[3]))

cv2.rectangle(frame, p1, p2, (255,0,0),2,1)

else :

# Tracking failure

            cv2.putText(frame,"Tracking failure detected", (100,80), cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,0,255),2)

# Display tracker type on frame

        cv2.putText(frame, tracker_type +" Tracker", (100,20), cv2.FONT_HERSHEY_SIMPLEX,0.75, (50,170,50),2);

# Display FPS on frame

        cv2.putText(frame,"FPS : " +str(int(fps)), (100,50), cv2.FONT_HERSHEY_SIMPLEX,0.75, (50,170,50),2);

# Display result

        cv2.imshow("Tracking", frame)

# Exit if ESC pressed

        k = cv2.waitKey(1) &0xff

        if k ==27 :break


c++版本

#include#include#includeusing namespace cv;using namespace std;// Convert to string#define SSTR( x ) static_cast< std::ostringstream & >( \( std::ostringstream() << std::dec << x ) ).str()int main(int argc, char **argv){// List of tracker types in OpenCV 3.2// NOTE : GOTURN implementation is buggy and does not work.string trackerTypes[6] = { "BOOSTING", "MIL", "KCF", "TLD","MEDIANFLOW", "GOTURN" };// vectortrackerTypes(types, std::end(types));// Create a trackerstring trackerType = trackerTypes[0];Ptr tracker;

#if (CV_MINOR_VERSION < 3)

{

tracker = Tracker::create(trackerType);

}

#else

{

if (trackerType == "BOOSTING")

tracker = TrackerBoosting::create();

if (trackerType == "MIL")

tracker = TrackerMIL::create();

if (trackerType == "KCF")

tracker = TrackerKCF::create();

if (trackerType == "TLD")

tracker = TrackerTLD::create();

if (trackerType == "MEDIANFLOW")

tracker = TrackerMedianFlow::create();

if (trackerType == "GOTURN")

tracker = TrackerGOTURN::create();

}

#endif

// Read video

VideoCapture video(0);

// Exit if video is not opened

if (!video.isOpened())

{

cout << "Could not read video file" << endl;

return 1;

}

// Read first frame

Mat frame;

bool ok = video.read(frame);

// Define initial boundibg box

Rect2d bbox(287, 23, 86, 320);

// Uncomment the line below to select a different bounding box

bbox = selectROI(frame, false);

// Display bounding box.

rectangle(frame, bbox, Scalar(255, 0, 0), 2, 1);

imshow("Tracking", frame);

tracker->init(frame, bbox);

while (video.read(frame))

{

// Start timer

double timer = (double)getTickCount();

// Update the tracking result

bool ok = tracker->update(frame, bbox);

// Calculate Frames per second (FPS)

float fps = getTickFrequency() / ((double)getTickCount() - timer);

if (ok)

{

// Tracking success : Draw the tracked object

rectangle(frame, bbox, Scalar(255, 0, 0), 2, 1);

}

else

{

// Tracking failure detected.

putText(frame, "Tracking failure detected", Point(100, 80), FONT_HERSHEY_SIMPLEX, 0.75, Scalar(0, 0, 255), 2);

}

// Display tracker type on frame

putText(frame, trackerType + " Tracker", Point(100, 20), FONT_HERSHEY_SIMPLEX, 0.75, Scalar(50, 170, 50), 2);

// Display FPS on frame

putText(frame, "FPS : " + SSTR(int(fps)), Point(100, 50), FONT_HERSHEY_SIMPLEX, 0.75, Scalar(50, 170, 50), 2);

// Display frame.

imshow("Tracking", frame);

// Exit if ESC pressed.

int k = waitKey(1);

if (k == 27)

{

break;

}

}

}


如果出现类似这样的错误是因为没有吧caffe模型放进去。模型在这里面https://github.com/opencv/opencv_extra/tree/master/testdata


下载之后在testdata\tracking\下面有goturn.caffemodel,和goturn.protxt把这个解压之后放到VS的工程目录下面就行。如果是python ,同理,放于工程目录下就行。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 151,688评论 1 330
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 64,559评论 1 273
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 101,749评论 0 226
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 42,581评论 0 191
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 50,741评论 3 271
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 39,684评论 1 192
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,122评论 2 292
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 29,847评论 0 182
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 33,441评论 0 228
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 29,939评论 2 232
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,333评论 1 242
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 27,783评论 2 236
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,275评论 3 220
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 25,830评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,444评论 0 180
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 34,553评论 2 249
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 34,618评论 2 249

推荐阅读更多精彩内容