美文网首页
Opencv 3.4 目标跟踪 goturn算法 实现

Opencv 3.4 目标跟踪 goturn算法 实现

作者: Fantesla | 来源:发表于2017-12-29 00:02 被阅读0次

    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 ,同理,放于工程目录下就行。

    相关文章

      网友评论

          本文标题:Opencv 3.4 目标跟踪 goturn算法 实现

          本文链接:https://www.haomeiwen.com/subject/fapbgxtx.html