Commit 230cc3e0 authored by Alexey Spizhevoy's avatar Alexey Spizhevoy

Minor fix in stitching_detailed

parent a7d053f1
...@@ -555,7 +555,7 @@ int main(int argc, char* argv[]) ...@@ -555,7 +555,7 @@ int main(int argc, char* argv[])
else if (seam_find_type == "gc_color") else if (seam_find_type == "gc_color")
{ {
#ifndef ANDROID #ifndef ANDROID
if (try_gpu) if (try_gpu && gpu::getCudaEnabledDeviceCount() > 0)
seam_finder = new detail::GraphCutSeamFinderGpu(GraphCutSeamFinderBase::COST_COLOR); seam_finder = new detail::GraphCutSeamFinderGpu(GraphCutSeamFinderBase::COST_COLOR);
else else
#endif #endif
...@@ -564,7 +564,7 @@ int main(int argc, char* argv[]) ...@@ -564,7 +564,7 @@ int main(int argc, char* argv[])
else if (seam_find_type == "gc_colorgrad") else if (seam_find_type == "gc_colorgrad")
{ {
#ifndef ANDROID #ifndef ANDROID
if (try_gpu) if (try_gpu && gpu::getCudaEnabledDeviceCount() > 0)
seam_finder = new detail::GraphCutSeamFinderGpu(GraphCutSeamFinderBase::COST_COLOR_GRAD); seam_finder = new detail::GraphCutSeamFinderGpu(GraphCutSeamFinderBase::COST_COLOR_GRAD);
else else
#endif #endif
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment