Updated CLI for GPU samples
This commit is contained in:
@@ -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));
|
||||
|
||||
|
Reference in New Issue
Block a user