Updated CLI for GPU samples

This commit is contained in:
Alexey Spizhevoy
2011-10-24 08:34:15 +00:00
parent 5656a9dd6b
commit a2090a44db
10 changed files with 247 additions and 131 deletions

View File

@@ -76,11 +76,16 @@ GpuMat d_result[2];
// CPU result
Mat result;
void printHelp()
{
std::cout << "Usage: driver_api_stereo_multi_gpu --left <left_image> --right <right_image>\n";
}
int main(int argc, char** argv)
{
if (argc < 3)
if (argc < 5)
{
std::cout << "Usage: stereo_multi_gpu <left_image> <right_image>\n";
printHelp();
return -1;
}
@@ -104,19 +109,27 @@ int main(int argc, char** argv)
}
// Load input data
Mat left = imread(argv[1], CV_LOAD_IMAGE_GRAYSCALE);
Mat right = imread(argv[2], CV_LOAD_IMAGE_GRAYSCALE);
if (left.empty())
Mat left, right;
for (int i = 1; i < argc; ++i)
{
std::cout << "Cannot open '" << argv[1] << "'\n";
return -1;
}
if (right.empty())
{
std::cout << "Cannot open '" << argv[2] << "'\n";
return -1;
if (string(argv[i]) == "--left")
{
left = imread(argv[++i], CV_LOAD_IMAGE_GRAYSCALE);
CV_Assert(!left.empty());
}
else if (string(argv[i]) == "--right")
{
right = imread(argv[++i], CV_LOAD_IMAGE_GRAYSCALE);
CV_Assert(!right.empty());
}
else if (string(argv[i]) == "--help")
{
printHelp();
return -1;
}
}
// Init CUDA Driver API
safeCall(cuInit(0));