Vegetation Segmentation Based On Variational Level Set Using Multi-Channel Local Wavelet Texture and Color
Vegetation Segmentation Based On Variational Level Set Using Multi-Channel Local Wavelet Texture and Color
Abstract The existing spectrum index-based methods tomation applications. DVC methods are generally based
for detecting vegetation coverage suffer from an overde- on visible spectral indexes (VSIs), including the Excess
pendence on spectrum. To address these issues, this pa- Green index (ExG) [1], Excess Red index (ExR) [2],
per proposes a graph cut based variational level set seg- Color Index of Vegetation Extraction (CIVE) [3], Ex-
mentation algorithm that combines multi-channel local cess Green minus excess Red index (ExGR) [4] and the
wavelet texture (MCLWT) and color. First, the prior Vegetative index (VEG) [5]. The advantages of VSI-
color is generated by automatic estimation based on based methods are that they can accurately segment
the mathematical morphology with a color histogram. vegetation, soil and sky and that visible spectral sen-
Then, local wavelet texture features are extracted using sors are not very expensive. Other DVC methods based
a multi-scale-and-orientation Gabor wavelet transfor- on multi-spectral images have advantages of wider cov-
mation followed by local median and entropy filtering. erage and better accuracy than VSIs, although only at
Next, in addition to the energy of color, that of M- considerably higher cost. However, both the VSI-based
CLWT is integrated into the variational level set model and multi-spectral-based methods share a common de-
based on kernel density estimation. Consequently, all ficiency: they have an over-dependence on spectrum or
energies are integrated into the graph cut based vari- color while neglecting wider spatial information such as
ational level set model. Finally, the proposed energy texture.
functional is made convex to obtain a global optimal Therefore, combinations of spectral indexes and oth-
solution, and a primal-dual algorithm with global rela- er methods have been proposed to improve the accura-
beling is adopted to accelerate the evolution of the lev- cy of DVC. Ponti [6] presented a combination of spec-
el sets. A comparison of the segmentation results from tral indexes and mean shift (MS) to segment balloon-
our proposed algorithm and other state-of-the-art al- captured remote sensing images, including MS+CIVE,
gorithms showed that our algorithm effectively reduces MS+ExG and MS+VVI. To improve the robustness of
the over-dependence on color and yields more accurate DVC in the presence of illumination variations or plan-
results in detecting vegetation coverage. t canopy shadows, Bai et al. [7] used particle swarm
Keywords Variational level set · detecting vegetation optimization (PSO) clustering and morphology mod-
coverage · local wavelet texture · graph cut elling to segment vegetation from soil in color images
acquired by an off-the-shelf digital camera affixed to an
image acquisition device. Unsupervised methods such
1 Introduction as MS or PSO clustering help in determining an ap-
propriate threshold for spectral indexes from images,
Detection of vegetation coverage (DVC) is a highly im- although errors always occur when the spectra of the
portant aspect of machine vision-based agricultural au- foreground and background overlap.
Zhun Fan
In addition to spectrum information or color, other
Department of Electrical Engineering, Shantou University, cues such as texture and shape have been considered in
Guangdong, P. R. China DVC. Feng et al. [8] used colors in the RGB color space
E-mail: [email protected] and textures based on the gray-level co-occurrence ma-
2 Tiejun Yang et al.
Fig. 1: Flow chart of the proposed method, including pre-processing, segmentation and post-processing.
level set model, the features are integrated as external b) Next, a mathematical morphology thicken oper-
energies into the graph cut based level set segmentation ator is applied, making fobj shk one pixel thicker.
model. To achieve global optimization more convenient- c) Finally, a mathematical morphology open oper-
ly, the proposed energy functional is transformed into a ator is imposed to reserve the most likely areas
convexity functional using Chan’s convexity transform of vegetation using a structuring element (SE),
method [15] as described above. We also adopt PDGRL which is a line 10 pixels long at a 2-degree angle.
to speed up the level set evolution in our method. Fi- The result is denoted by fobj .
nally, in the post-processing stage, mathematical mor- 3) To obtain the AEPC for non-vegetation, we use the
phology operations are used to erase tiny islands and following steps:
fill in small holes in the segmentation results. a) A shrink operator is applied to fbw .
b) Then, a mathematical morphology dilate opera-
tor is used to find the likely areas of non-vegetation
3.1 Automatic estimation of prior color using an SE line 8 pixels long at a 2-degree angle.
The result is denoted by fbkg .
To obtain prior color, manual labelling is widely used in 4) Using fobj and fbkg as masks [14], color histograms
[18,14]. However, manual labelling requires human in- of the vegetation hf and non-vegetation hb are cal-
teraction and its efficiency is low—especially for large culated.
data sets. To address this problem, automatic estima- 5) The histograms are smoothed by a Gaussian filter
tion of prior color (AEPC) is required. Here, we pro- and then linearly interpolated at each pixel’s inten-
pose an approach based on CIVE and mathematical sity value from the original image, yielding the final
morphology that can automatically generate approxi- prior color histograms, Hf and Hb , respectively.
mate probability distributions of vegetation and non-
vegetation. CIVE is used here to generate estimated Fig. 2 shows an example of AEPC. The visualized image
segmentation results. Then, mathematical morphology of AEPC fAEP C in Fig. 2h is composed as follows:
is applied to obtain the primary components of the ob-
jects and background. These processes are described fAEP C = [R, G, B] ,
(3)
below: R = fbkg , G = 0, B = fobj .
1) A given image f (x, y) is first segmented using CIVE
to obtain binary segmentation results. CIVE is com-
puted as follows:
3.2 Multi-channel local wavelet texture
fCIV E = 0.441R − 0.811G + 0.385B + 18.787, (2)
where R, G and B are the values of the color chan- Because vegetation usually has different textural fea-
nels of f (x, y) in the RGB color space. Then Otsu tures than those of surrounding soil, water, etc., we
thresholding is used to transform fCIV E into a bi- introduce a texture feature called multi-channel local
nary value, fbw . wavelet texture (MCLWT) into the proposed model for
2) To obtain the AEPC for vegetation, we use the fol- vegetation segmentation. A Gabor wavelet transform
lowing steps: is insensitive to illumination variation and geometric
a) A mathematical morphology shrink operator is transformation. Therefore, we use it here to transfor-
applied to the inverse of fbw , shrinking the areas m the original image into a multi-resolution frequency
of vegetation to connected lines or points. The domain. The two-dimensional Gabor wavelet transform
results are denoted by fobj shk . w (·, ·) for an image f (x, y) with a frequency of f0 and
4 Tiejun Yang et al.
(a)
(b)
Acknowledgements This research was partly supported by 14. M. Unger, T. Pock, and H. Bischof. Global relabeling
the Guangdong Provincial Key Laboratory of Digital Signal for continuous optimization in binary image segmenta-
and Image Processing Techniques (2013GDDSIPL-03) and by tion. Energy Minimization Methods in Computer Vi-
the Guangxi Basic Ability Promotion Project for Young and sion and Pattern Recognition, Lecture Notes in Comput-
Middle-aged Teachers (2017KY0247). er Science, 6819:104–117, 2011.
15. Nikolova M Chan T. F., Esedoglu S. Algorithms for
finding global minimizers of image segmentation and de-
noising models. SIAM Journal on Applied Mathematics,
References 66(5):1632–1648, 2006.
16. Tiejun Yang, Lin Huang, and Ping Hu. Vegetation seg-
1. Von Bargen K. Woebbecke D.M., Meyer G.E. and mentation based on level set using local wavelet texture.
Mortensen D.A. Color indices for weed identification un- In 2016 IEEE 13th International Conference on Signal
der various soil, residue, and lighting conditions. Trans- Processing, pages 222–227.
actions of the ASAE (USA), 38(1):259–269, 1995. 17. Xavier Bresson, Selim Esedoglu, Pierre Vandergheynst,
2. George E. Meyer, Timothy W. Hindman, and Koppolu Jean-Philippe Thiran, and Stanley Osher. Fast global
Laksmi. Machine vision detection parameters for plant minimization of the active contour/snake model. Jour-
species identification. Photonics East, 1999. nal of Mathematical Imaging and Vision, 28(2):151–167,
3. T. Kataoka, T. Kaneko, H. Okamoto, and S. Hata. Crop 2007.
growth estimation system using machine vision. In Ad- 18. Carsten Rother, Vladimir Kolmogorov, and Andrew
vanced Intelligent Mechatronics, 2003. AIM 2003. Pro- Blake. Grabcut: interactive foreground extraction us-
ceedings. 2003 IEEE/ASME International Conference ing iterated graph cuts. Acm Transactions on Graphics,
on, pages b1079–b1083. 23(3):309–314, 2004.
4. George E. Meyer and Joao Camargo Neto. Verification 19. Tiejun Yang, Lin Huang, and Chuanxian Jiang. Wavelet
of color vegetation indices for automated crop imaging and total variation based level set segmentation method
applications. Computers and Electronics in Agriculture, for portraits. International Journal of Digital Content
63(2):282–293, 2008. Technology and its Applications, 7(4):973–982, 2013.
5. T. Hague, N. Tillett, and H. Wheeler. Automated crop 20. Chan T. F. and Vese L. A. Active contours without edges.
and weed monitoring in widely spaced cereals. Precision IEEE Transactions on Image Processing, 10(2):266–277,
Agriculture, 7(1):21–32(12), 2006. 2001.
6. M.P.Ponti. Segmentation of low-cost remote sensing im- 21. Zhenping Xie, Shitong Wang, and Dewen Hu. New in-
ages combining vegetation indices and mean shift. IEEE sight at level set & gaussian mixture model for natu-
Geoscience and Remote Sensing Letters, 10(1):67–70, ral image segmentation. Signal Image Video Processing,
2013. 7(3):521–536, 2013.
7. Xiaodong Bai, Zhiguo Cao, Yu Wang, Zhenghong Yu, 22. J. Chang and J.W. Fisher. Efficient mcmc sampling
Zhu Hu, Xuefen Zhang, and Cuina Li. Vegetation seg- with implicit shape representations. In IEEE Conference
mentation robust to illumination variations based on on Computer Vision and Pattern Recognition (CVPR),
clustering and morphology modelling. Biosystems En- pages 2081–2088, 2011.
gineering, 125(3):80–97, 2014. 23. Yizong Cheng. Mean shift, mode seeking clustering.
8. Quanlong Feng, Jiantao Liu, and Jianhua Gong. Uav re- IEEE Transactions on Pattern Analysis and Machine
mote sensing for urban vegetation mapping using random Intelligence, 17(8):790–799, 1995.
forest and texture analysis. Remote Sensing, 7(1):1074– 24. C. Gée, J. Bossu, G. Jones, and F. Truchetet. Crop/weed
1094, 2015. discrimination in perspective agronomic images. Com-
9. S. Aksoy, H. G. Akcay, and T. Wassenaar. Automat- puters and Electronics in Agriculture, 60(1):49–59, 2008.
ic mapping of linear woody vegetation features in agri- 25. J. C. Neto. A Combined Statistical-Soft Computing Ap-
cultural landscapes using very high resolution imagery. proach for Classification and MappingWeed Species in
IEEE Transactions on Geoscience and Remote Sensing, Minimum-Tillage Systems. Thesis, 2006.
48(1):511–522, 2010. 26. Lei F. Tian and David C. Slaughter. Environmentally
adaptive segmentation algorithm for outdoor image seg-
10. M. S. Nia, Zhe Wang, P. Gader, S. A. Bohlman, S. J.
mentation. Computers and Electronics in Agriculture,
Graves, and M. Petrovic. Impact of atmospheric correc-
21(3):153–168, 1998.
tion and image filtering on hyperspectral classification of
27. L. M. Bergasa, M. Mazo, A. Gardel, M. A. Sotelo, and
tree species using support vector machine. Journal of
L. Boquete. Unsupervised and adaptive gaussian skin-
Applied Remote Sensing, 9(1):095990, 2015.
color model. Image and Vision Computing, 18(12):987–
11. Osher S. and Sethian J. Fronts propagating with cur-
1003, 2000.
vature dependent speed: Algorithms based on hamilton-
28. M. J. Jones and J. M. Rehg. Statistical color models
jacobi formulations. Journal of Computational Physics,
with application to skin detection. International Journal
79(1):12–49, 1988.
of Computer Vision, 46(1):81–96, 2002.
12. Kangwei Liu, Jianhua Wan, and Zhongzhi Han. Ab-
normal event detection and localization using level set
based on hybrid features. Signal Image Video Process-
ing, 11(6):1–7, 2017.
13. Zahra Shahvaran, Kamran Kazemi, and Mo-
hamad Sadegh Helfroush. Simultaneous vector-valued
image segmentation and intensity nonuniformity correc-
tion using variational level set combined with markov
random field modeling. Signal Image Video Processing,
10(5):887–893, 2016.