1 Introduction

In wireless sensor networks for the situation understanding [1], different modalities are needed such as radar and optical images. In [2], it was shown that radar and images are two independent modalities which can be treated independently. In this paper, we focus on image modality in wireless sensor networks. The visibility and contrast of the captured images may be affected and may be degraded by many reasons, such as poor environmental condition, camera sensor noise, and other uncertain factors [3, 4]. It is a necessary step before further processing and understanding the images to improve image quality with enhancement and denoising algorithm in many vision applications [5, 6]. The contrast and detail of images can be improved remarkably by those algorithms, so it will be easier for human or machine to identify and understand.

The image enhancement algorithms can be divided into two categories: one is spatial domain algorithm, which enhances the images by recalculating the gray value of the pixels. The other is transform domain algorithm, which enhances the images by adjusting the image coefficients in some transform domain.

At present, the multi-scale analysis theories (MAT) have been widely used in transform domain image enhancement algorithms. The main idea of MAT is to enhance the image details in different scales and directions. The key points of these algorithms are the selection of multi-scale analysis tools and coefficients enhancement models.

In image processing applications, the most well-known multi-scale analysis tools are wavelet [7], contourlet [8], ridgelet [9], and other transform methods. Shearlet transform is a new multi-scale analysis tool which integrates the advantages of wavelet and contourlet [10]. It is well adapted to represent anisotropic features. Shearlet can provide an optimally sparse representation for multi-dimensional data [11], so it suits well for image processing tasks. The traditional multi-scale transforms tend to introduce pseudo-Gibbs artifacts owing to the down sampling operations. To eliminate this phenomenon, the shift-invariant non-subsampled transforms were proposed. Non-subsampled shearlet transform (NSST) can capture the geometrical structure of an image, such as edges, much more precisely than those traditional multi-scale transforms. The decomposition procedure for NSST is similar to non-subsampled contourlet transform (NSCT). But the computational cost and complexity of NSST is much lower than that of NSCT [12, 13]. And the number of directions for shearing has no limitation in the process of NSST, while the number of directions for NSCT decomposing must be any power of 2. So NSST has more flexibility in the description of the geometric structure of the image.

We apply NSST to image enhancement algorithm in this paper. The shearlet coefficients are processed by a non-linear function which can suppress the noise coefficients and can strengthen the weak detail coefficients. Since the thresholds of the function are selected adaptively, the coefficients of each scale and direction are enhanced. The enhanced results of the proposed algorithm show a good performance on both the enhancement of the image details and the suppression of noise.

2 Image enhancement based on NSST

2.1 Enhancement function and its properties

The aim for enhancing images is to improve the quality of the images so that image perceptions required by further processes can be obtained. So a well-designed enhancement algorithm can amplify the details of image, can adjust the image contrast, and meanwhile, can remove the noise of signals. It means that the coefficients of the image should be processed separately. So it is a good choice to use the non-linear function as the enhancement model.

The enhancement model used in our enhancement algorithm is proposed by A.F. Laine in 1996 [14]. Since then, many scholars have applied this function to image enhancement processing and have achieved good results.

$$ f(x)= a\left[\mathrm{sigm}\left( c\left( x- b\right)\right)-\mathrm{sigm}\left(- c\left( x+ b\right)\right)\right] $$
(1)

Where \( a=\frac{1}{\mathrm{sigm}\left( c\left(1- b\right)\right)-\mathrm{sigm}\left(- c\left(1+ b\right)\right)} \), sigm is defined as \( \mathrm{sigm}(x)=\frac{1}{1+{e}^{- x}} \)

There are two parameters named c and b in this function, where b is used to control the enhancement range, and its value range is (0, 1); while c is used to control the enhancement intensity, which usually takes the fixed value between 20 and 50. In the existing researches, many enhancement algorithms took the selection of the two parameters of b and c as key factors to improve the non-linear enhancement effect. Figure 1 demonstrates the curves with different b and c values in different colors. As shown in Fig. 1, the shape of the enhancement function varies greatly with different b and c values. At the same time, it also can be found that (take the positive direction of X axis as an example) two special points play important roles in the form of the function curve. One point is the first intersection between the curve and the line f(x) = x. It determines whether the image coefficient should be suppressed by denoising or enhanced by amplifying. It means that the horizontal coordinate of the point is the boundary of denoising and enhancement. The other important point is the intersection between the function curve and line y = 1, it determines the extent of the enhancement. The coefficients with greater value than the horizontal coordinate of the point will be increased to the maximum value of the coefficients, while the smaller coefficients will be amplified non-linearly according to the function curve.

Fig. 1
figure 1

The comparison of the function curves with different value of b and c

The two important points on the function curve, in this paper, are proposed as two thresholds to determine the form of the enhancement function. The thresholds will be selected adaptively according to the image multi-resolution decomposition coefficients of different scales and directions.

2.2 Selection of the thresholds

We set two thresholds of \( {T_k^l}_{\mathrm{low}} \) and \( {T_k^l}_{\mathrm{high}} \) in the enhancement function. \( {T_k^l}_{\mathrm{low}} \) is the threshold to distinguish the coefficients between the enhancement set and the suppression set, \( {T_k^l}_{\mathrm{high}} \) is used to control the contrast stretching degree of the enhanced coefficients. The two thresholds correspond to the two special points in the enhancement curves analyzed in 2.1.

Shearlet transform can decompose images into different scales and directions. And the coefficients of different scale have different features. The high-frequency coefficients often correspond to the image edges and details. This kind of coefficient needs obvious enhancement, but noise signals also mostly exist in the high-frequency coefficients. So it is required to suppress the high-frequency coefficients which have smaller absolute values. The low-frequency coefficients correspond to the image background or smooth parts, such coefficients should not be sharply enhanced but should try to maintain its contrast. Since there is less noise signals in low-frequency coefficients, their demand for denoising is not obvious. Therefore, we will set the thresholds for different decomposition scales and directions adaptively.

\( {T_k^l}_{\mathrm{low}} \) corresponds to the horizontal coordinate of the enhancement function curve at the point of x = f(x). It is supposed that \( {T_k^l}_{\mathrm{low}} \) is proportional to the standard deviation of the decomposition coefficients, and the standard deviation calculation function satisfies formula (2).

$$ \sigma =\sqrt{\frac{1}{\mathrm{MN}}{\displaystyle \sum_{m=1}^M{\displaystyle \sum_{n=1}^N{\left({c}_k^l\left( m, n\right)-\mathrm{mea}{\mathrm{n}}_c\right)}^2}}} $$
(2)

Where σ is the standard deviation of the coefficients; \( {c}_k^l\left( m, n\right) \) is the value of the coefficient in the position of (m, n); meanc is the mean value of the coefficients of k scale l direction.

The frequency of decomposition coefficient rises with the increase of the decomposition scales. Noise signals are more likely to be gathered in the high-scale coefficients. So the calculation of \( {T_k^l}_{\mathrm{low}} \) needs to consider the scale factors. The calculation function of \( {T_k^l}_{\mathrm{low}} \) is shown as formula (3).

$$ {T_k}_{\mathrm{low}}^l=\frac{1}{8}\sigma *\mathrm{level} $$
(3)

Where \( {T_k^l}_{\mathrm{low}} \) represents the first threshold of the coefficients in k scale l direction, σ is the standard deviation of the decomposition coefficients, level is the decomposition scale. Formula (3) means that the value of \( {T_k^l}_{\mathrm{low}} \) is propositional to the decomposition scale. That is, the greater the value of the decomposition scale, the more stringent the coefficients denoising.

\( {T_k}_{\mathrm{high}}^l \) corresponds to the value of \( \frac{9}{c}+ b \) because from this point the derivative of the enhancement function begins to approach zero [15]. Its calculation function satisfies formula (4)

$$ {T_k}_{\mathrm{high}}^l={T}_{\mathsf{low}}^l* n/\mathrm{leve}{\mathrm{l}}^2 $$
(4)

Where \( {T_k}_{\mathrm{high}}^l \) represents the second threshold of the coefficients in k scale l direction; n is a constant, it is assigned to 150 in our experiments; \( {T_k}_{\mathrm{high}}^l \) is inversely proportional to the decomposition scale. That is, the higher the frequency of the decomposition coefficient, the stronger the enhancement degree of the coefficient. While for the low-frequency coefficient, the enhancement curve tends to be gentle.

2.3 The overall framework of the algorithm

A flow chart of the proposed enhancement algorithm is illustrated in Fig. 2, which is composed of seven steps:

  1. (1)

    Decompose the original image by NSST to K levels and L orientations.

  2. (2)

    Normalize the two highest frequency scales of coefficients in order to prepare them to be enhanced.

  3. (3)

    Calculate the standard variance of the normalized coefficients in each layer and direction.

  4. (4)

    Calculate the two enhancement thresholds for each layer and direction and use the thresholds to determine the form of the enhancement function.

  5. (5)

    Enhance the normalized coefficients by the enhancement function.

  6. (6)

    Anti-normalize the enhanced coefficients.

  7. (7)

    Recompose the image by the enhanced high-frequency coefficients and the original low-frequency coefficients.

Fig. 2
figure 2

Flow chart of the proposed algorithm

Our algorithm is suitable to enhance grayscale images. Since it can suppress noise signals by the first threshold, it can also be used to enhance noisy images.

3 Experiment results and discussion

A set of grayscale images is used to demonstrate the performance of the proposed algorithm. The images include the standard 8-bit grayscale images of Lena, Barbara, and an infrared image. The hardware platform of the experiment is a desktop computer with 3.2 GHz CPU and 4 G RAM.

NSST, NSCT, wavelet are selected as image multi-resolution decomposition tools. The enhancement effects and evaluation parameters are shown in Fig. 3 and Tables 1, 2, and 3. Among the four images of each set, the first one is the original image, the second one is enhanced by wavelet soft threshold enhancement algorithm, the third one is enhanced by NSCT adaptive enhancement algorithm [16], and the last one is enhanced by the proposed algorithm.

Fig. 3
figure 3

The performance of different enhancement algorithms on grayscale images

Table 1 Objective evaluation parameters for Lena image set
Table 2 Objective evaluation parameters for Barbara image set
Table 3 Objective evaluation parameters for the infrared image set

By contrast, we can find that the performance of NSST and NSCT is much better than that of wavelet by human eyes, especially at the detail parts. We choose three evaluation parameters to evaluate the enhancement methods. They are contrast improvement index (CII) [17], edge preservation index (EPI) [18], and running time. The objective evaluation results listed in Tables 1, 2, and 3 show that the proposed algorithm improves the image contrast and maintains the details. The running time of the program is acceptable.

An infrared image taken under haze environment is used to measure the ability of the algorithms to suppress noise while enhancing details. We choose equivalent number of looks (ENL) [19] as an extra evaluation parameters to testify the algorithms’ ability to suppress noise. It can be seen from Fig. 4 and Table 4 that the proposed algorithm has the smallest ENL and biggest CII and EPI among the three. So it means that our algorithm has good enhancement effect on noisy images.

Fig. 4
figure 4

The performance of different enhancement algorithms on noisy image

Table 4 Objective evaluation parameters for the noisy image set

4 Conclusions

NSST can be used as a promising method in image processing tasks, such as denosing, fusion, enhancement. We propose a non-linear enhancement algorithm in wireless sensor networks. Two thresholds of the enhancement function are selected adaptively according to the standard variation of the coefficients and the decomposition scale. Experiment results show that our method performs well on the enhancement of both standard gray images and infrared image. And it can also be used to enhance noisy images since it has a good ability of suppress noises among signals.