Opencv 3.4 目標跟蹤 goturn算法 實現(xiàn)

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

這里是一篇問題匯總。

對于VS2017在按照相關(guān)配置,把opencv的和opencontrib源碼下載編譯之后,配置好opencv后(配置請參考上一篇文章)

對于python 在anaconda上的有opencv3.3.1的contrib編譯好的,https://www.lfd.uci.edu/~gohlke/pythonlibs/到這里下載就行注意版本

如果是opencv3.4,需要重新編譯python版本的,試了一下失敗了。

下面是算法實現(xiàn)。

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;

}

}

}


如果出現(xiàn)類似這樣的錯誤是因為沒有吧caffe模型放進去。模型在這里面https://github.com/opencv/opencv_extra/tree/master/testdata


下載之后在testdata\tracking\下面有g(shù)oturn.caffemodel,和goturn.protxt把這個解壓之后放到VS的工程目錄下面就行。如果是python ,同理,放于工程目錄下就行。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 227,572評論 6 531
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,071評論 3 414
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,409評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,569評論 1 307
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,360評論 6 404
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 54,895評論 1 321
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 42,979評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,123評論 0 286
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,643評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,559評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,742評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,250評論 5 356
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 43,981評論 3 346
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,363評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,622評論 1 280
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,354評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,707評論 2 370

推薦閱讀更多精彩內(nèi)容