opencv/opencv_contrib
1
0
Fork 0
mirror of https://github.com/opencv/opencv_contrib.git synced 2025-10-19 02:16:34 +08:00
Code Issues Releases Wiki Activity

added first version of hpe_humaneva dataset loader

Browse Source
This commit is contained in:
Dmitriy Anisimov
2014-10-25 11:46:08 +04:00
parent 44f47427df
commit 38168334f6
3 changed files with 418 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
modules/datasets/include/opencv2/datasets/hpe_humaneva.hpp Normal file
View File

@@ -0,0 +1,85 @@
/*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_HPE_HUMANEVA_HPP
#define OPENCV_DATASETS_HPE_HUMANEVA_HPP
#include <string>
#include <vector>
#include "opencv2/datasets/dataset.hpp"
#include <opencv2/core.hpp>
namespace cv
{
namespace datasets
{
struct HPE_humanevaObj : public Object
{
char person; // 1..4
std::string action;
int type1;
std::string type2;
Matx13d ofs;
std::string fileName;
std::vector<std::string> imageNames; // for HumanEva_II
};
enum datasetType
{
humaneva_1 = 1,
humaneva_2
};
class CV_EXPORTS HPE_humaneva : public Dataset
{
public:
virtual void load(const std::string &path) = 0;
static Ptr<HPE_humaneva> create(int num=humaneva_1);
};
}
}
#endif

100
modules/datasets/samples/hpe_humaneva.cpp Normal file
View File

@@ -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/hpe_humaneva.hpp"
#include <opencv2/core.hpp>
#include <cstdio>
#include <cstdlib> // atoi
#include <string>
#include <vector>
using namespace std;
using namespace cv;
using namespace cv::datasets;
int main(int argc, char *argv[])
{
const char *keys =
"{ help h usage ? | | show this message }"
"{ path p |true| path to dataset folder }";
CommandLineParser parser(argc, argv, keys);
string path(parser.get<string>("path"));
if (parser.has("help") || path=="true")
{
parser.printMessage();
return -1;
}
for (unsigned int i=1; i<=2; ++i)
{
printf("\tHumanEva %u\n", i);
char number[2];
sprintf(number, "%u", i);
string pathCurr(path+"HumanEva_"+number+"/");
Ptr<HPE_humaneva> dataset = HPE_humaneva::create(i);
dataset->load(pathCurr);
// ***************
// dataset contains pair of rgb\dep images
// For example, let output train size and last element.
HPE_humanevaObj *example = static_cast<HPE_humanevaObj *>(dataset->getTrain().back().get());
printf("train size: %u\n", (unsigned int)dataset->getTrain().size());
printf("last train video:\n");
printf("person: %u\n", example->person);
printf("action: %s\n", example->action.c_str());
printf("type1: %u\n", example->type1);
printf("type2: %s\n", example->type2.c_str());
printf("filename: %s\n", example->fileName.c_str());
printf("num images: %u\n", (int)example->imageNames.size());
printf("ofs:");
for (int j=0; j<3; ++j)
{
printf(" %f", example->ofs(0, j));
}
printf("\n");
}
return 0;
}

233
modules/datasets/src/hpe_humaneva.cpp Normal file
View File

@@ -0,0 +1,233 @@
/*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/hpe_humaneva.hpp"
#include "opencv2/datasets/util.hpp"
#include "precomp.hpp"
#include <cstring>
namespace cv
{
namespace datasets
{
using namespace std;
class HPE_humanevaImp : public HPE_humaneva
{
public:
HPE_humanevaImp() {}
//HPE_humanevaImp(const string &path);
virtual ~HPE_humanevaImp() {}
virtual void load(const string &path);
private:
void loadDataset(const string &path);
};
/*HPE_humanevaImp::HPE_humanevaImp(const string &path)
{
loadDataset(path);
}*/
void HPE_humanevaImp::load(const string &path)
{
loadDataset(path);
}
void HPE_humanevaImp::loadDataset(const string &path)
{
train.push_back(vector< Ptr<Object> >());
test.push_back(vector< Ptr<Object> >());
validation.push_back(vector< Ptr<Object> >());
for (unsigned int i=1; i<=4; ++i)
{
char number[2];
sprintf(number, "%u", i);
string pathDatasetI(path + "S" + number + "/Image_Data/");
string pathDatasetS(path + "S" + number + "/Sync_Data/");
vector<string> fileNames;
getDirList(pathDatasetI, fileNames);
for (vector<string>::iterator it=fileNames.begin(); it!=fileNames.end(); ++it)
{
string &file = *it;
vector<string> elems;
split(file, elems, '_');
if (elems.size() != 3)
{
continue;
}
Ptr<HPE_humanevaObj> curr(new HPE_humanevaObj);
curr->person = (char)i;
curr->action = elems[0];
curr->type1 = atoi(elems[1].c_str());
curr->fileName = pathDatasetI+file;
unsigned int type2End = 2;
if (elems[2][type2End+1] != ')')
{
type2End = 3;
}
curr->type2 = elems[2].substr(1, type2End);
file = file.substr(0, file.length()-3) + "ofs";
ifstream infileOFS((pathDatasetS + file).c_str());
string line;
unsigned int j = 0;
while (getline(infileOFS, line))
{
curr->ofs(0, j) = atof(line.c_str());
++j;
}
train.back().push_back(curr);
}
}
}
//
// HumanEva II
//
class HPE_humanevaImpII : public HPE_humaneva
{
public:
HPE_humanevaImpII() {}
//HPE_humanevaImpII(const string &path);
virtual ~HPE_humanevaImpII() {}
virtual void load(const string &path);
private:
void loadDataset(const string &path);
};
/*HPE_humanevaImpII::HPE_humanevaImpII(const string &path)
{
loadDataset(path);
}*/
void HPE_humanevaImpII::load(const string &path)
{
loadDataset(path);
}
void HPE_humanevaImpII::loadDataset(const string &path)
{
train.push_back(vector< Ptr<Object> >());
test.push_back(vector< Ptr<Object> >());
validation.push_back(vector< Ptr<Object> >());
for (unsigned int i=1; i<=2; ++i)
{
char number[2];
sprintf(number, "%u", i*2); // 2 & 4
string pathDatasetI(path + "S" + number + "/Image_Data/");
string pathDatasetS(path + "S" + number + "/Sync_Data/");
vector<string> fileNames;
getDirList(pathDatasetI, fileNames);
for (vector<string>::iterator it=fileNames.begin(); it!=fileNames.end(); ++it)
{
string &file = *it;
vector<string> elems;
split(file, elems, '_');
if (elems.size() != 3)
{
continue;
}
Ptr<HPE_humanevaObj> curr(new HPE_humanevaObj);
curr->person = (char)i;
curr->action = elems[0];
curr->type1 = atoi(elems[1].c_str());
curr->fileName = pathDatasetI+file;
unsigned int type2End = 2;
if (elems[2][type2End+1] != ')')
{
type2End = 3;
}
curr->type2 = elems[2].substr(1, type2End);
vector<string> imageNames;
getDirList(curr->fileName, imageNames);
for (vector<string>::iterator itI=imageNames.begin(); itI!=imageNames.end(); ++itI)
{
string &image = *itI;
if (image.substr(image.length()-3) == "png")
{
curr->imageNames.push_back(image);
}
}
file = file.substr(0, file.length()) + ".ofs";
ifstream infileOFS((pathDatasetS + file).c_str());
string line;
unsigned int j = 0;
while (getline(infileOFS, line))
{
curr->ofs(0, j) = atof(line.c_str());
++j;
}
train.back().push_back(curr);
}
}
}
Ptr<HPE_humaneva> HPE_humaneva::create(int num)
{
if (humaneva_2==num)
{
return Ptr<HPE_humanevaImpII>(new HPE_humanevaImpII);
}
return Ptr<HPE_humanevaImp>(new HPE_humanevaImp);
}
}
}