Abstract
GraphCut algorithm has shown its effectiveness when solving many computer vision tasks. However, its heavy computational nature makes it hard to apply in real-world applications. Many attempts have been made to accelerate GraphCut algorithm, most successfully seen in methods that utilize parallel computing platforms like CUDA. In this paper, we introduce a parallel implementation of push relabel algorithm for GraphCut on CUDA designed for the image stitching problem. Furthermore, we propose global suppression heuristic to boost the convergence process of the algorithm. Experiment results on sets of thermal infrared and RGB images show that our method can be up to 3 times faster than the fastest sequential algorithm while obtaining satisfactory stitched images. Our source code will be soon available for further research.
Similar content being viewed by others
Data availability
All data generated or analysed during this study are included in this published article [and its supplementary information files].
References
Kolmogorov, V., Zabih, R.: What energy functions can be minimized via graph cuts? IEEE Trans. Pattern Anal. Mach. Intell. 26(2), 147–159 (2004)
Li, Y., Sun, J., Shum, H.Y.: ACM SIGGRAPH 2005 papers on—SIGGRAPH ’05. ACM Press (2005)
Boykov, Y., Kolmogorov, V.: An experimental comparison of min-cut/max-flow algorithms for energy minimization in vision. IEEE Trans. Pattern Anal. Mach. Intell. 26(9), 1124–1137 (2004)
Kwatra, V., Schödl, A., Essa, I., Turk, G., Bobick, A.: Graphcut textures: image and video synthesis using graph cuts. ACM Trans. Graph. 22(3), 277–286 (2003)
Kolmogorov, V., Zabih, R.: In: Proceedings Eighth IEEE International Conference on Computer Vision. ICCV (2001)
Ford, L.R., Fulkerson, D.R.: Flows in Networks. Princeton University Press (1963)
Edmonds, J., Karp, R.M.: Theoretical improvements in algorithmic efficiency for network flow problems. J. ACM 19(2), 248–264 (1972)
Goldberg, A.V.: Algorithmic Aspects in Information and Management, pp. 212–225. Springer, Berlin Heidelberg (2009)
Vineet, V., Narayanan, P.J.: In: 2008 IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops, pp. 1–8 (2008)
Hussein, M., Varshney, A., Davis, L.: On implementing graph cuts on cuda
Dixit, N., Keriven, R., Paragios, N.: Gpu-cuts: Combinatorial optimisation, graphic processing units and adaptive object extraction (2005)
Goldberg, A.V., Tarjan, R.E.: A new approach to the maximum-flow problem. J. ACM 35(4), 921–940 (1988)
Shekhovtsov, A., Hlavác, V.: A distributed mincut/maxflow algorithm combining path augmentation and push-relabel. CoRR arXiv:abs/1109.1146 (2011)
Dinic, E.A.: Algorithm for solution of a problem of maximum flow in a network with power estimation. Sov. Math. Doklady 11, 1277–1280 (1970)
Kohli, P., Torr, P.: Dynamic graph cuts for efficient inference in Markov random fields. IEEE Trans. Pattern Anal. Mach. Intell. 29(12), 2079–2088 (2007)
Liu, J., Sun, J.: In: 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition, IEEE (2010)
Cherkassky, B.V., Goldberg, A.V.: On implementing the push-relabel method for the maximum flow problem. Algorithmica 19(4), 390–410 (1997)
Anderson, R.J., Setubal, J.C.: In: Proceedings of the fourth annual ACM symposium on Parallel algorithms and architectures—SPAA ’92. ACM Press (1992)
Bader, D., Sachdeva, V.: A cache-aware parallel implementation of the push-relabel network flow algorithm and experimental evaluation of the gap relabeling heuristic. ISCA PDCS (2005)
Cormen, T.H., Leiserson, C.E., Rivest, R.L., Stein, C.: Introduction to Algorithms, Third Edition, MIT Press (2009)
Funding
Not applicable.
Author information
Authors and Affiliations
Contributions
TN and MB conducted analyses and proposed the global suppression heuristic and early stopping strategy. TN, TN, THT, and MB participated in constructing the code base for experiments. TN provided experiment results in terms of figures and tables. MB wrote the main manuscript text. HN participated in the design of the study. All authors reviewed the manuscript.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no competing interests.
Consent to participate
Not applicable.
Consent for publication
Not applicable.
Ethical approval
Not applicable.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Bui, M., Nguyen, T., Ninh, H. et al. Global suppression heuristic: fast GraphCut in GPU for image stitching. SIViP 17, 2671–2678 (2023). https://doi.org/10.1007/s11760-023-02483-5
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11760-023-02483-5