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)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容