Added VOT2015 Dataset interface + VOT2015 sample cpp

9 years ago · a38145dcec
parent d41639a949
commit a38145dcec
3 changed files with 433 additions and 0 deletions
--- a/modules/datasets/include/opencv2/datasets/track_vot.hpp
+++ b/modules/datasets/include/opencv2/datasets/track_vot.hpp
@ -0,0 +1,102 @@
+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                           License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2014, Itseez Inc, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of the copyright holders may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Itseez Inc or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#ifndef OPENCV_DATASETS_TRACK_VOT_HPP
+#define OPENCV_DATASETS_TRACK_VOT_HPP
+
+#include <string>
+#include <vector>
+#include <map>
+#include <iostream>
+
+#include "opencv2/datasets/dataset.hpp"
+#include "opencv2/datasets/util.hpp"
+#include <opencv2/highgui.hpp>
+
+#include <opencv2/core.hpp>
+#include <sys/stat.h>
+
+using namespace std;
+
+namespace cv
+{
+namespace datasets
+{
+
+//! @addtogroup datasets_track
+//! @{
+
+struct TRACK_votObj : public Object
+{
+    int id;
+    std::string imagePath;
+	vector <Point2d> gtbb;
+};
+
+class CV_EXPORTS TRACK_vot : public Dataset
+{
+public:
+	static Ptr<TRACK_vot> create();
+
+    virtual void load(const std::string &path) = 0;
+
+	virtual int getDatasetsNum() = 0;
+
+	virtual int getDatasetLength(int id) = 0;
+
+	virtual bool initDataset(int id) = 0;
+
+	virtual bool getNextFrame(Mat &frame) = 0;
+
+	virtual vector <Point2d> getGT() = 0;
+
+protected:
+	vector <vector <Ptr<TRACK_votObj> > > data;
+	int activeDatasetID;
+	int frameCounter;
+};
+
+//! @}
+
+}
+}
+
+#endif
--- a/modules/datasets/samples/track_vot.cpp
+++ b/modules/datasets/samples/track_vot.cpp
@ -0,0 +1,100 @@
+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                           License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2014, Itseez Inc, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of the copyright holders may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Itseez Inc or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#include "opencv2/datasets/track_vot.hpp"
+
+#include <opencv2/core.hpp>
+#include <opencv2/highgui.hpp>
+#include <opencv2/imgproc.hpp>
+
+#include <cstdio>
+#include <cstdlib>
+
+#include <string>
+#include <vector>
+#include <set>
+
+using namespace std;
+using namespace cv;
+using namespace cv::datasets;
+
+#define DATASET_ID 1
+
+int main(int argc, char *argv[])
+{
+	const char *keys =
+		"{ help h usage ? |    | show this message }"
+		"{ path p         |true| path to folder with dataset }";
+	CommandLineParser parser(argc, argv, keys);
+	string path(parser.get<string>("path"));
+	if (parser.has("help") || path == "true")
+	{
+		parser.printMessage();
+		getchar();
+		return -1;
+	}
+
+	Ptr<TRACK_vot> dataset = TRACK_vot::create();
+	dataset->load("D:/opencv/VOT 2015");
+	printf("Datasets number: %d\n", dataset->getDatasetsNum());
+	for (int i = 1; i <= dataset->getDatasetsNum(); i++)
+		printf("\tDataset #%d size: %d\n", i, dataset->getDatasetLength(i));
+
+	dataset->initDataset(DATASET_ID);
+
+	for (int i = 0; i < dataset->getDatasetLength(DATASET_ID); i++)
+	{
+		Mat frame;
+		dataset->getNextFrame(frame);
+
+		//Draw Ground Truth BB
+		vector <Point2d> gtPoints =  dataset->getGT();
+		for (int j = 0; j < (int)(gtPoints.size()-1); j++)
+			line(frame, gtPoints[j], gtPoints[j + 1], Scalar(0, 255, 0), 2);
+		line(frame, gtPoints[0], gtPoints[(int)(gtPoints.size()-1)], Scalar(0, 255, 0), 2);
+
+		imshow("VOT 2015 DATASET TEST...", frame);
+		waitKey(100);
+	}
+
+	getchar();
+	return 0;
+}
--- a/modules/datasets/src/track_vot.cpp
+++ b/modules/datasets/src/track_vot.cpp
@ -0,0 +1,231 @@
+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                           License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2014, Itseez Inc, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of the copyright holders may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Itseez Inc or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#include "opencv2/datasets/track_vot.hpp"
+
+using namespace std;
+
+namespace cv
+{
+namespace datasets
+{
+
+class TRACK_votImpl : public TRACK_vot
+{
+public:
+	//Constructor
+    TRACK_votImpl()
+	{
+
+		activeDatasetID = 1;
+		frameCounter = 0;
+	}
+	//Destructor
+    virtual ~TRACK_votImpl() {}
+
+	//Load Dataset
+    virtual void load(const string &path);
+
+	virtual int getDatasetsNum();
+
+	virtual int getDatasetLength(int id);
+
+	virtual bool initDataset(int id);
+
+	virtual bool getNextFrame(Mat &frame);
+
+	virtual vector <Point2d> getGT();
+
+private:
+    void loadDataset(const string &path);
+
+    string numberToString(int number);
+};
+
+void TRACK_votImpl::load(const string &path)
+{
+    loadDataset(path);
+}
+
+string TRACK_votImpl::numberToString(int number)
+{
+	string out;
+    char numberStr[9];
+    sprintf(numberStr, "%u", number);
+    for (unsigned int i=0; i<8-strlen(numberStr); ++i)
+    {
+        out += "0";
+    }
+    out += numberStr;
+	return out;
+}
+
+inline bool fileExists(const std::string& name)
+{
+	struct stat buffer;
+	return (stat(name.c_str(), &buffer) == 0);
+}
+
+void TRACK_votImpl::loadDataset(const string &rootPath)
+{
+	ifstream namesList(rootPath + "/list.txt");
+	//ifstream lengthsList(rootPath + "/lengths.txt");
+	vector <int> datasetsLengths;
+	string datasetName;
+
+	if (namesList.is_open())
+	{
+		int currDatasetID = 0;
+
+		//All datasets/folders loop
+		while (getline(namesList, datasetName))
+		{
+			currDatasetID++;
+			vector <Ptr<TRACK_votObj> > objects;
+
+			//All frames/images loop
+			Ptr<TRACK_votObj> currDataset(new TRACK_votObj);
+
+			//Open dataset's ground truth file
+			ifstream gtList(rootPath + "/" + datasetName + "/groundtruth.txt");
+			if (!gtList.is_open())
+			cout << "Error to open groundtruth.txt!!!";
+
+			//Make a list of datasets lengths
+			int currFrameID = 1;
+			if (currDatasetID == 0)
+			cout << "VOT 2015 Dataset Initialization...\n";
+
+			do
+			{
+				currFrameID++;
+				string fullPath = rootPath + "/" + datasetName + "/" + numberToString(currFrameID) + ".jpg";
+				if (!fileExists(fullPath))
+					break;
+
+				//Make VOT Object
+				Ptr<TRACK_votObj> currObj(new TRACK_votObj);
+				currObj->imagePath = fullPath;
+				currObj->id = currFrameID;
+
+				//Get Ground Truth data
+				double	x1 = 0, y1 = 0,
+						x2 = 0, y2 = 0,
+						x3 = 0, y3 = 0,
+						x4 = 0, y4 = 0;
+				string tmp;
+				getline(gtList, tmp);
+				sscanf(tmp.c_str(), "%lf,%lf,%lf,%lf,%lf,%lf,%lf,%lf", &x1, &y1, &x2, &y2, &x3, &y3, &x4, &y4);
+				currObj->gtbb.push_back(Point2d(x1, y1));
+				currObj->gtbb.push_back(Point2d(x2, y2));
+				currObj->gtbb.push_back(Point2d(x3, y3));
+				currObj->gtbb.push_back(Point2d(x4, y4));
+
+				//Add object to storage
+				objects.push_back(currObj);
+
+			} while (true);
+
+			datasetsLengths.push_back(currFrameID-1);
+			data.push_back(objects);
+		}
+	}
+	else
+	{
+		cout << rootPath + "Couldn't find a *list.txt* in VOT 2015 folder!!!";
+	}
+
+	namesList.close();
+	return;
+}
+
+int TRACK_votImpl::getDatasetsNum()
+{
+	return data.size();
+}
+
+int TRACK_votImpl::getDatasetLength(int id)
+{
+	if (id > 0 && id <= (int)data.size())
+		return data[id - 1].size();
+	else
+	{
+		cout << "Dataset ID is out of range...\n " << "Allowed IDs are: 1~" << (int)data.size() << endl;
+		return -1;
+	}
+}
+
+bool TRACK_votImpl::initDataset(int id)
+{
+	if (id > 0 && id <= (int)data.size())
+	{
+		activeDatasetID = id;
+		return true;
+	}
+	else
+	{
+		cout << "Dataset ID is out of range...\n " << "Allowed IDs are: 1~" << (int)data.size() << endl;
+		return false;
+	}
+}
+
+bool  TRACK_votImpl::getNextFrame(Mat &frame)
+{
+	frame = imread(data[activeDatasetID - 1][frameCounter]->imagePath);
+	frameCounter++;
+	return !frame.empty();
+}
+
+Ptr<TRACK_vot> TRACK_vot::create()
+{
+    return Ptr<TRACK_votImpl>(new TRACK_votImpl);
+}
+
+vector <Point2d> TRACK_votImpl::getGT()
+{
+	Ptr <TRACK_votObj> currObj = data[activeDatasetID - 1][frameCounter - 1];
+	return currObj->gtbb;
+}
+
+}
+}
+
+