Person re-identification, one of the most challenging tasks in the field of computer vision, aims to recognize the same person cross different cameras. The local feature information has been proved that can improve performance efficiently. Image horizontal even division and pose estimation are two popular methods to extract the local feature. However, the former may cause misalignment, the latter needs much calculation. To fill this gap and improve performance, an efficient strategy is proposed in this work. First, a joint uneven channel information network consisting of an uneven channel information extraction network and a channel information fusion network is designed. Different from the traditional image division, the former can divide images horizontally and unevenly with strong alignment based on weak pose estimation, and extract multiple channel information. The latter can joint channel information based on channel validity and generate an efficient similarity descriptor. To optimize the joint uneven channel information network efficiently, this work proposes a blend metric loss. The extra image information is utilized to dynamically adjust the penalty for the sample distance and the distance margin based on the outlier of the hardest sample to construct i-TriHard loss. Besides, softmax loss and center loss are embedded in the blend metric loss, which can guide the network to learn more discriminative features. Our method achieves 89.6% mAP and 95.9% Rank-1 on Market-1501, 79.9% mAP and 89.4% Rank-1 on DukeMTMC. The proposed method also performs excellently on occluded datasets.
Hinweise
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Introduction
With the popularity of video surveillance, person re-identification technology has attracted more and more attention. Given an image of a person taken by a specific camera, it can retrieve other images of this person from the gallery taken by other cameras. With the wide application in practical monitoring and intelligent security fields, it has always been a mainstream subject.
Traditional methods use manually designed features, which have achieved results, but they are still far from being applied in actual scenes. With the quick development of deep learning, person re-identification has also made rapid progress. Studies of person re-identification based on deep learning can be divided into two branches: representation learning and metric learning. Representation learning does not directly consider the similarity between images when training the network, but treats the person re-identification as a classification task or a verification task. Metric learning aims to learn the similarity of two images through the network. In the person re-identification, the similarity between different images from the same person is greater than that of different images from different persons.
Anzeige
Most of researches concentrate on extracting the global feature [1, 2]. Newer researches attempt to divide the image into small pieces [3] and utilizes pose estimation to extract different body regions [4, 5]. However, the former may cause misalignment, the latter needs much calculation. To fill this gap, a joint uneven channel information network (JUCIN) consisting of an uneven channel information extraction network and a channel information fusion network is proposed. The former, different from the traditional horizontal even division, a weak pose estimation module combining the advantages of even division and pose estimation is utilized to divide image into four uneven local channels with strong alignment, and consider the original image as a global channel. Therefore, five channel information can be generated. Next, the latter fuses five pairs of channel information into the similarity descriptor, and a dynamic fusion strategy based on channel validity is embedded in the pipeline. Last, this descriptor is used to finish person re-identification task.
Contrastive loss [6], triplet loss [7], TriHard loss [8], quadruplet loss [9], and margin sample mining loss [10] are proposed to optimize the neural network. It is different from representation learning in that metric learning relies on sample pairs. In this work, to optimize the feature extraction networks in JUCIN, a blend metric loss (BML) is proposed. First, TriHard loss is improved to construct i-TriHard loss, which can utilize the extra image information to dynamically adjust the penalty for the sample distance and the distance margin, thereby optimizing the spatial distribution between positive samples and negative samples. Besides, softmax loss and center loss [11] are embedded in the blend metric loss, which can guide the network to learn more discriminate features.
The proposed method shows excellent performance when compared to state-of-the-art methods. The contributions of this work can be summarized as follows: (1) A weak pose estimation strategy is proposed to guide horizontal uneven division, this allows the module to achieve channel alignment efficiently. (2) A joint uneven channel information network based on weak pose estimation is proposed to extract and fuse channel information dynamically. (3) A blend metric loss is proposed to optimize network, which can optimize the spatial distribution between samples to enhance the performance.
Related work
In the related researches of representation learning, He et al. [12] used a convolutional neural network to extract the global feature. Additionally, Lin et al. [13] introduced different attribute features such as hair and clothes to improve generalization. Varior et al. [14] divided the images into several pieces, and then input these pieces into a long short-term memory network to generate the final features. However, this method requires images had strong alignment. Zhao et al. [15] proposed a Spindle Net, utilizing a convolutional pose machines to locate landmarks and extract different body regions. Then, it could fuse the global features and the local features. Zheng et al. [16] used a pose estimation network and an affined transformation to divide the image into several areas, and then used a PoseBox to correct the local areas. The local areas and the original image were then input into the network to extract the features. Wei et al. [17] introduced the global–local-alignment descriptor to reduce the negative impact of the changeable pose. The image was divided into three areas, and the global average pooling method could solved the mismatch of the image. The fusion feature was then extracted. Zheng et al. [18] designed a SP distance adaptive alignment model, which could align local features without additional information, and used a dynamic alignment algorithm to find the shortest path. Yu et al. [19] proposed a deep discriminative representation method to learn the features and impose a discriminative constraint on the feature representation.
Anzeige
In the related researches of metric learning, Varior et al. [6] proposed a contrast loss and used it in the twin network, which could dynamically adjust the distance threshold according. Schroff et al. [7] proposed a triplet loss. The network took the triple image as input, which could pull the distance between positive sample pairs and push the distance between negative sample pairs, efficiently improving feature discrimination and re-identification performance. Hermans et al. [8] proposed a TriHard loss. For each sample batch, the loss selected a positive sample and a negative sample that were the most difficult to distinguish, and then constructed a sample set to calculate the final loss. Chen et al. [9] proposed a quadruple loss, which took four different images as the input of the network. Compared to triplet loss, quadruple loss considered the absolute distance between positive samples and negative samples. Zhu et al. [20] designed a hybrid similarity function to measure the similarity between the feature pairs, they also proposed a deep hybrid similarity learning method, which could reasonably allocate the complexity of feature learning and measurement learning, thereby improving the performance. Liu et al. [21] proposed a nonlinear deep metric learning strategy based on deep belief networks and component analysis, which used a data change methods to maximize the number of images and learned nonlinear feature mapping. Din et al. [22] improved the loss function and optimized the learning algorithm of network. They proposed a deep neural network with scalable distance-driven feature learning, which could effectively improve the performance.
Fig. 1
The pipeline of JUCIN. It consists of an uneven channel information extract networks (UCIEN) and a channel information fusion networks (CIFN). WPEM refers to weak pose estimation module. GCI refers to global channel information. LCI refers to local channel information. BML refers to blend metric loss. WDM refers to weight decision module. IFM refers to information fusion module
×
Joint uneven channel information network
A joint uneven channel information network (JUCIN) is proposed in Fig. 1, which consists of an uneven channel information extract network and a channel information fusion network. First, the image is input to a weak pose estimation module, which can locate five landmarks and divide the image into four different pieces unevenly. Different from the traditional horizontal division, this strategy can achieve image division with alignment. Different pieces are defined as different channels in this work, and then utilize convolutional neural networks to extract the channel information. To optimize the feature extraction networks, a blend metric loss is proposed, the detailed information will be explained in Sect. 4. A channel information fusion network is proposed to fuse different channel information from two compared images. Different channel information should contribute differently. Therefore, fusion weight which is generated by a weight decision module based on channel validity is introduced. Last, the weight and the channel information are combined to generate the final similarity descriptor for person re-identification.
Uneven channel information extraction network
The uneven channel information extraction network can extract a global channel information and four uneven local channel information. As the core of the extraction network, the weak pose estimation module can divide the image horizontally and unevenly with strong alignment. The details are illustrated in the following sections.
Weak pose estimation
As mentioned above, the local feature information can improve the performance. The popular method to extract the local feature is dividing images into several pieces evenly or generating some local regions based on pose estimation.
However, the former requires the image pair to have strong alignment, or else this method will cause extra negative influence due to misalignment as shown in Fig. 2. The latter can solve misalignment but needs much additional computation.
Fig. 2
Even division causes misalignment, uneven division achieves alignment
×
To balance these two methods and improve the traditional image division method, a novel division strategy which use a weak pose estimation module is proposed. It can combine the advantages from horizontal even division and pose estimation, achieving unevenly but efficient image division without much calculation.
The pipeline of weak pose estimation module is shown in Fig. 3. For an input image, it first detects five landmarks: top of head, neck, crotch, left foot, and right foot, and then divides the image into four local channels horizontally and evenly.
Fig. 3
The pipeline of weak pose estimation module
×
Inspired by CPM [23], the weak pose estimation module utilizes a similar mechanism to locate different landmarks. But in this work, five landmarks are enough to divide the images. Therefore, some stages and neural layers are reduced.
After determining five landmarks, the following equations are utilized to generate four local channels:
where input is the matrix of the input image, width is the image width, \(L_i\) is the ith landmark, and y(\(L_i\)) is the height coordinate of the landmark.
In summary, the weak pose estimation module utilizes \(L_1\), \(L_4\), and \(L_5\) to generate \(c_1\), utilizes \(L_1\) and \(L_2\) to generate \(c_2\), utilizes \(L_2\) and \(L_3\) to generate \(c_3\), and utilizes \(L_3\), \(L_4\), and \(L_5\) to generate \(c_4\). Based on this, the input image can be divided horizontally and unevenly into four different channels, and make these channels as local channels. The traditional division method may cause misalignment, but this novel method combines the simplicity of horizontal division and the efficiency of pose estimation. It can accomplish the image division with less computation and resolve the misalignment between the image pairs.
Uneven channel information extraction
In this work, an uneven channel information extraction network is proposed to extract the channel information. As illustrated in Fig. 1, it consists of five branches.
The first branch takes the original image as input, which is used to extract the global channel information. The second branch takes the processed image consisting of three pieces as input, which can reduce the negative influence of background and extract the local channel information. The last three branches take three horizontally divided images as input, which are all part of the original image and do not overlap. Finally, three local channel information can be generated.
The pipeline of information extraction in each branch is shown in Fig. 4, and the core module is ResNet50. The batch normalization (BN) layer can constrain the distribution of the features and balance the dimensions of different features.
Fig. 4
The pipeline of channel information extraction
×
Therefore, for an image, five different channel information can be extracted, including a global channel information and four uneven local channel information.
For the image pair to be compared, the weak pose estimation module gives the corresponding channel strong alignment, which can solve the misalignment problem between image pieces without increasing much calculation. the uneven channel information extraction network can also make the channel information more robust.
In summary, the uneven channel information extraction network based on a weak pose estimation combines image horizontal division and pose estimation, which can achieve efficient alignment between image pieces. It aims to learn a global channel information and four local channel information. The pipeline can be divided into five branches in structure and function. An input image can generate five feature information through this pipeline. This can be summarized as follows:
where f\(^{\ c_{0}}\) is the global channel information, f\(^{\ c_{1}}\), f\(^{\ c_{2}}\), f\(^{\ c_{3}}\), and f\(^{\ c_{4}}\) are four local channel information.
Channel information fusion network
Five channel information can be obtained through the uneven channel information extraction network. Dealing with this information and combining them into an efficient descriptor is extremely important. A channel information fusion network is proposed to accomplish this task. As shown in Fig. 1, it consists of a weight decision module and an information fusion module. The former is the core of the fusion network and can generate different fusion weights based on channel validity. The latter can fuse the channel information similarity with the fusion weight to generate the final similarity descriptor.
Fig. 5
An example of channel information fusion. Assume the global channel information similarity is \({s}=0.7\), two channel validities are \({v}_p=1.0\) and \({v}_g=1.0\), and the kernel function is average. Therefore, the weight of global channel information is \({w} = 1.0\). The similarity and weight for other channels are calculated by the same way. This work uses the weighted sum to generate the final similarity
×
For an image pair consisting of a probe image and a gallery image, it can obtain five channel information pairs. Then, handling these information pairs to generate a similarity descriptor should be focused. In this work, different contributions from channels to final similarity should be weighted. To accomplish this, a weight decision module based on channel validity is embedded to generate the channel weight dynamically. Finally, the channel information fusion module fuses dynamic weights and channel information into the similarity descriptor. The similarity between the probe image and the gallery image can be presented as follows:
$$\begin{aligned} S = {\sum \limits _{i=0}^{N-1}{w_{i}*s(f_{p}^{c_{i}},f_{g}^{c_{i}})}}, \end{aligned}$$
(6)
where N is the number of channels, \(w_i\) is the weight of ith channel, s(\(\cdot \)) stands for the similarity between the ith channel information from the probe image and the gallery image. f\(_{p}^{c_{i}}\) and f\(_{g}^{c_{i}}\) are the ith channel information from the probe image and the gallery image.
The core of the channel information fusion network is determining the weight from different channels. In this pipeline, a weight decision module is designed. The following section will introduce the working mechanism.
The local channel information is considered as the supplementary information for the global channel information in this work, the former is more important than other channel information. Because there are some deviations in the generation of the local channels with the deviation of the weak pose estimation module. Therefore, the weight from the global channel is fixed to 1, and the weights from four local channels are less than or equal to 1. From this, the concept of channel validity is proposed, which is defined as the ratio of the channel height to the original image height. It can be formulated as follows:
where H\(_{c_{i}}\) is the height of the ith channel, H is the height of the the original image.
Since horizontal division is utilized to generate four local channels, the height of the local channel is the height of image piece. This work considers the channel validity to be related to the piece scale. If the horizontal piece scale is larger, the channel validity is larger and the channel weight is larger. In other words, if the piece is larger, the corresponding channel information will be more important to the final similarity.
For the image pair consisting of the probe image and the gallery image, two corresponding channel validities from the identical channels can be calculated. The weight decision module takes the channel validity pair as input. A kernel function is utilized to generate the channel weight. The channel information fusion network combines five different channel weights with the channel similarity to generate the final similarity descriptor. Therefore, Eq. (6) can be transformed as follows:
$$\begin{aligned} S = {\sum \limits _{i=0}^{N-1}{\varphi (v_{p}^{i},v_{g}^{i})\times s(f_{p}^{c_{i}},f_{g}^{c_{i}})}}, \end{aligned}$$
(8)
where \(\varphi \)(\(\cdot \)) is kernel function.v\(_{p}^{i}\) and v\(_{g}^{i}\) are the ith channel validity from the probe image and the gallery image.
Figure 5 shows an example of channel information fusion. Five pairs of channel similarity information and channel weights are combined to generate the final similarity.
The fusion similarity generated by the information fusion module is considered the discriminative similarity descriptor for person re-identification in this work. A higher fusion similarity indicates that two compared images have a higher similarity.
Blend metric loss
To optimize the feature extraction networks efficiently in JUCIN, a blend metric loss (BML) is proposed in Fig. 6, which consists of the improved i-TriHard loss, softmax loss, and center loss. The dynamic penalty is imposed on TriHard loss based on extra image information to construct i-TriHard loss, which is the core of BML and can optimize the spatial distribution of the samples. The strategy for dynamic distance margin adjustment is also embedded in this pipeline. BML can guide the network to learn more discriminate features in the embedding space. The following sections show the details.
Fig. 6
The pipeline of blend metric loss
×
Improved i-TriHard loss
Triplet loss [7] is widely used in metric learning, and a large number of subsequent losses are derived from that, which can guide the network to learn more discriminative features and improve robustness. Triplet loss takes a positive sample pair and a negative sample pair as input, which are a (anchor), p (positive sample), and n (negative sample). The positive sample pair is a and p, while the negative sample pair is a and n. Triplet loss aims to maximize the distance between the negative samples and minimize the distance between the positive samples. Triplet loss is formulated as follows:
where d(\(\cdot \)) is Euclidean distance, a\(_i\) is the ith anchor, p\(_i\) is the ith positive sample, n\(_i\) is the ith negative sample, and m is the distance margin.
As an evolution of triplet loss, TriHard loss [8] assumes the positive samples and the negative samples in triplet loss are easy to distinguish, which is not conducive to network training. By selecting the farthest positive sample and the nearest negative sample to optimize the network, TriHard loss can enhance the generalization and the representation of the network. TriHard loss is formulated as follows:
where P is the number of identities, K is the number of image from each identity.
TriHard loss uses the hardest positive sample and the hardest negative sample exclusively, assigning the same weight to the hardest sample for each anchor. However, it ignores the remaining positive samples and negative samples, and it cannot consider the specific distribution of the samples in the training batch.
Ideally, if the performance of the network is efficient, all positive samples or negative samples should be clustered together. If there is a large gap between the hardest positive sample and other positive samples, or there is a large gap between the hardest negative sample and other negative samples, it means that the penalty of the network needs to be further strengthened. Therefore, TriHard loss is improved to construct i-TriHard loss which can dynamically adjust the penalty for sample distance and distance margin. A specific description is illustrated in Fig. 7. Different penalties are formulated based on the distribution of the samples.
Fig. 7
The mechanism of the improved i-TriHard loss. a is the anchor, the green points are positive samples, the red points are negative samples, p\(_1\) and p\(_2\) are the two hardest positive samples in two different batches, n\(_1\) and n\(_2\) are the two hardest negative samples in two different batches. The distribution divergence of p\(_2\) is larger than p\(_1\) since its outlier is larger. Therefore, a greater penalty should be imposed on p\(_2\). Analogously, the distribution divergence of n\(_2\) is larger than n\(_1\), its outlier is larger, a greater penalty should be imposed on n\(_2\). The number of arrows indicates the penalty degree
×
The proposed i-TriHard loss can be formulated as follows:
where w\(_p\) and w\(_n\) are two penalty weights for the distance of the hardest positive sample and the hardest negative sample, and w\(_m\) is the weight of the distance margin. They are generated dynamically based on the distribution of the samples. In other words, if the hardest positive sample has a larger outlier, w\(_p\) should be larger, and if the hardest negative sample has a larger outlier, w\(_n\) should be smaller. This leads to a larger loss and increases network penalty. This mechanism can guide the networks to learn a better embedding space.
In this work, the ratio between the distance of the hardest positive sample or the hardest negative sample to the average positive distance or the average negative distance is utilized to describe the outlier degree, and considering it as the penalty coefficient of the hardest distance. These can be formulated as follows:
where P(a) and N(a) are all positive samples and all negative samples in a batch, N\(_p\) and N\(_n\) is the number of the positive samples and the negative samples in a batch.
By these definitions, w\(_p\) always satisfies w\(_p \ge \)1 and w\(_n\) always satisfies w\(_n \le \)1. These weights can improve efficiency in early stages of training. As the training progresses, w\(_p\) slowly decreases and w\(_p\) slowly increases, which can gradually lower the penalty and it is conducive to the follow-up training.
The proposed i-TriHard loss is always greater than the original TriHard loss because of the new dynamic weights for the hardest positive sample distance and the hardest negative sample distance. If the loss remains a large value all the time, it will make the network training more difficult and result in underfitting. To solve this problem, the distance margin between the positive samples and the negative samples is also dynamically adjusted. This can appropriately reduce the distance threshold and the loss. The weight of this distance margin can be formulated as follows:
Softmax loss is the most popular classification loss which aims to construct several hyperplanes for dividing embedding space. It can be formulated as follows:
where y\(_i=1\) when the label of ith sample is the true label, otherwise, y\(_i=0\). s\(_i\) is the output of softmax.
As triplet loss cannot achieve optimal constraint at the global level, the combination with softmax loss can guide neural network to learn more discriminative features.
Besides, triplet loss and its variants only consider the distance difference between the hardest positive sample and the hardest negative sample. Although the discreteness between different classes is considered, it ignores the absolute value of the distance and the feature cohesion. Center loss [11] can solve this problem efficiently, it learns the center of each class and penalizes the loss through the distance between the features and the class center. Center loss is formulated as follows:
where m is the batch size, x\(_i\) is the feature corresponding to the ith sample, c\(_{y_{i}}\) represents the feature center of the class y\(_i\).
In this work, i-TriHard loss, softmax loss, and center loss are combined to construct the blend metric loss, which can be formulated as follows:
$$\begin{aligned} L_\mathrm{BML} = L_{\text{ i }-TriHard}+L_\mathrm{softmax}+L_\mathrm{center}. \end{aligned}$$
(18)
The dynamic adjustment for the hardest sample distance and the distance margin is embedded in the blend metric loss, and the combination of different loss is utilized to optimize the network, which can guide the network to learn more discriminative features.
Experiments
Datasets
Market-1501 [24] is collected from Tsinghua University, using six cameras to collect 32,668 images from 1501 different persons. The training set consists of 12,936 images from 751 persons, and the testing set consists of a query set containing 3368 images and a gallery set containing 19,732 images. DukeMTMC [25] is collected from Duke University, using eight cameras to collect 36,411 images from 1404 different persons. The training set consists of 16,522 images from 702 persons, and the testing set consists of a query set containing 2228 images and a gallery set containing 17,661 images. Occluded-DukeMTMC [26] is an occluded dataset, using eight cameras to collect 35,489 images from 1404 different persons. The training set consists of 15,618 images from 702 persons, and the testing set consists of a query set containing 2210 images and a gallery set containing 17,661 images. Partial iLIDS [27] is a simulated partial dataset, including 238 images from 119 person identities. Each label contains an occluded image and a non-occluded image.
As the most popular evaluation metrics, cumulative matching characteristic (CMC) curves and mean average precision (mAP) are utilized to evaluate the performance of person re-identification model in this work.
Implementation details
The weak pose estimation module is trained based on MPII dataset [28] to generate five landmarks and four local channels. ResNet50 is the core module in the uneven channel information extraction network, and it is initialized with ImageNet [29] pre-trained model. The input image is resized to 256 \(\times \) 128. The batch size is set to 64, the number of different identities is set to 16, and the number of different images from per identity is set to 4. The maximum training epoch is set to 200. To prevent the overfitting, this work uses four different regularization strategies. First, the warmup learning rate is applied in our pipeline. The initial learning rate is 3 \(\times \) 10\(^{-6}\), and then rises to 3 \(\times \) 10\(^{-4}\), next drops to 3 \(\times \) 10\(^{-5}\) in the 50th epoch, and last drops to 3 \(\times \) 10\(^{-6}\) in the 100th epoch. Second, the random flipping and the random erasing strategy [43] are used to augmented the training data in our experiment. Third, dropout strategy is applied to inactivate certain nerve units, the rate of dropout is set as 0.8. Fourth, L2 regularization is used in the reverse optimization of the network. The whole experiment is implemented on the hardware equipped with a GeForce GTX 2080Ti GPU.
Comparison with state-of-the-arts
The proposed method is compared with state-of-the-arts in Table 1. Our method achieves 89.6% mAP and 95.9% Rank-1 in Market-1501, 79.9% mAP and 89.4% Rank-1 in DukeMTMC. Compared to the best method in the table, our method has different improvement on two datasets.
Table 1
Comparison with state-of-the-arts on Market-1501 and DukeMTMC
To further verify the effectiveness of our method, the occluded environment is also considered. The performance on Occluded-DukeMTMC and Partial-iLIDS datasets are shown in Tables 2 and 3.
Table 2
Comparison with state-of-the-arts on Occluded-DukeMTMC
In Table 2, our method achieves significant improvement compared to the best method HG [49], mAP is improved from 50.5 to 53.4% and Rank-1 is improved from 61.7 to 62.1%. In Table 3, our method has certain improvement compared to the best method LKWS [53], showing increase from 80.7 to 82.4% in Rank-1 and increase from 88.2 to 89.9% in Rank-3.
Fig. 8
Ablation evaluation on channel information fusion. a Market-1501. b DukeMTMC
×
From the data gathered, it can be concluded that our proposed method achieves advanced performance whether under a normal environments or an occluded environments. The data supports the effectiveness of the proposed method.
Ablation evaluation on channel information fusion
The proposed JUCIN framework can extract and joint multiple channel information. To prove the validity of the proposed fusion strategy, an ablation experiment based on different channel information is conducted. The result is illustrated in Fig. 8. “Global” represents the global channel information. “Local_1”, “Local_2”, “Local_3”, and “Local_4” represent four local channel information.
Compared to the global channel information, our method obtains 1.2% mAP and 0.9% Rank-1 gains on Market-1501. It also obtains 1.4% mAP and 1% Rank-1 gains on DukeMTMC. The result can be improved further with the introduction of more additional channel information. When all channel information is merged, further improved results of 89.6% mAP and 95.9% Rank-1 can be obtained in Market-1501, 79.9% mAP and 89.4% Rank-1 in Duke-MTMC. The data demonstrate the effectiveness of our proposed method of channel information fusion.
Evaluation on uneven division
Different from the traditional even division, the uneven division combing pose estimation is utilized in this work. To prove the effectiveness, a comparative experiment based on even division and pose estimation is conducted in Table 4.
Table 4
Evaluation on uneven division
Method
Market-1501
DukeMTMC
mAP
Rank-1
mAP
Rank-1
Even division (\(\times \)2)
88.0
94.1
77.1
86.4
Even division (\(\times \)3)
88.4
94.5
77.7
87.1
Even division (\(\times \)6)
88.8
94.9
77.9
87.6
Pose estimation (CPM)
88.5
94.7
78.0
88.1
Uneven division (Ours)
89.6
95.9
79.9
89.4
No matter what the value of even division number is, their performance is inferior to the uneven division. Compared with evenly dividing image into three pieces, 1.2% mAP gain can be obtained by the proposed method. If only using pose estimation to extract the local features, it can achieve 88.5% mAP and 78.0% mAP, which is lower than our method.
To further prove that our method can solve the problems of horizontal even division and pose estimation. This work compares our proposed method with other methods based on horizontal even division and pose estimation. The methods based on horizontal even division include HPM [54], MMHPN [55], LHF [56], and RANGE [57]. The methods based on pose estimation include DPA [58], HOReID [59], MBRAN [60], and DSA [61]. The result is illustrated in Table 5.
Table 5
Evaluation on horizontal even division and pose estimation
Obviously, compared with the methods based on horizontal even division and the methods based on pose estimation, our proposed method can obtain the best performance. Compared with the best methods listed in the table, the method proposed can obtain 2% mAP gain with 0.2% Rank-1 gain on Market-1501, and 1.7% mAP gain with 2.1% Rank-1 gain on DukeMTMC.This proves that our method can solve the problems of horizontal even division and pose estimation.
Ablation evaluation on blend metric loss. a Market-1501. b DukeMTMC
×
Evaluation on kernel function
The weight decision module takes the channel validity pair as input. A kernel function is utilized for processing input to generate the channel weight. Then, channel information fusion network combines five different channel weights with the channel similarity to infer the similarity. To evaluate the influence of different kernel functions on the performance, an experiment comparing different kernel functions is conducted. The results are shown in Table 6.
As shown in Table 5, different kernel functions achieve different performance. The first kernel function fixes the weight of each channel to 1. Equal-weight channel information fusion is performed in this condition, resulting in 88.1% mAP with 95.5% Rank-1 on Market-1501 and 78.1% mAP with 87.6% Rank-1 on DukeMTMC. Compared to others, the last kernel function has the best performance. It ignores the channel validity from gallery image, and considers the channel validity from the probe image as the channel weight. These results show that when testing, the channel information from the probe image should be paid more attention, as it is more significant.
Ablation evaluation on blend metric loss
The proposed blend metric loss consists of the improved i-TriHard loss, softmax loss and center loss. To verify the effectiveness of our loss, an ablation experiment based on different losses is conducted. The results are shown in Fig. 9.
Compared to softmax loss, gains of 3.9% mAP and 2.3% Rank-1 can be obtained in Market-1501. Gains of 3.7% mAP and 3.6% Rank-1 can be obtained in DukeMTMC. BML achieves 0.3% mAP with 0.3% Rank-1 gains and 0.4% mAP with 0.6% Rank-1 gains in two datasets compared to not using center loss. BML achieves 3.5% mAP with 1.9% Rank-1 gains and 3.4% mAP with 3.1% Rank-1 gains compared to not using i-TriHard loss. This shows that the introduction of i-TriHard loss and center loss can optimize network efficiently. That also verifies the effectiveness of our proposed blend metric loss.
Unlike the traditional TriHard loss, the proposed i-TriHard loss introduces dynamic adjustment to the hardest sample distance and the distance margin based on the distribution of extra images. To demonstrate the effectiveness of the improved strategy, the result from comparative experiments based on i-TriHard loss are illustrated in Table 7. TriHard is the traditional loss, “DP” means dynamically adjust the hardest positive sample distance, “DN” means dynamically adjust the hardest negative sample distance, and “DM” means dynamically adjust the distance margin.
Table 7
Evaluation on dynamic adjustment strategy
Dynamic adjustment
Market-1501
DukeMTMC
mAP
Rank-1
mAP
Rank-1
JUCIN + TriHard
86.9
94.7
77.0
86.7
JUCIN + TriHard + DP
88.0
95.1
79.1
87.7
JUCIN + TriHard + DN
88.3
95.1
78.9
87.6
JUCIN + TriHard + DP + DN
88.9
95.5
79.6
88.5
JUCIN + TriHard + DP + DN + DM
89.6
95.9
79.9
89.4
As shown in Table 7, traditional TriHard loss without dynamic adjustment can only obtain 86.9% mAP with 94.7% Rank-1 on Market-1501 and 77.0% mAP with 86.7% Rank-1 on DukeMTMC, which is lower than our proposed i-TriHard loss. After a dynamic penalty is embedded, the method can achieve 2% mAP and 2.6% mAP improvement. Dynamically adjusting the distance margin saw even further improvements of 0.7% mAP with 0.4% Rank-1 gains on Market1501 and 0.3% mAP with 0.9% Rank-1 gains on DukeMTMC.
To prove the effectiveness of this proposed i-TriHard loss, we add a new experiment applying TriHard loss and other improved variants of TriHard to train our network. The result is shown in Table 8. The compared losses include TriHard loss [8], Quadruplet loss [9], MSML [10], ALHSM [62]. Softmax loss and center loss are also combined with these losses to optimize the network.
As illustrated in Table 8, the proposed i-TriHard loss in this work can achieve the best performance improvement. Besides, compared with other variants of TriHard loss in the table, our designed loss does not need introduce more complex calculation and it is simple to implement, which proves the effectiveness of i-TriHard loss.
Conclusions
The joint uneven channel information network consisting of an uneven channel information extract network and a channel information fusion network is proposed in this work. Different from the traditional image division, the former can divide images horizontally and unevenly with strong alignment, and extract multiple channel information based on a weak pose estimation module, which combines the simplicity of horizontal division and the efficiency of pose estimation, and this demonstrates a novel image division strategy. The latter can joint channel information based on the channel validity and generate the efficient similarity descriptor. To optimize the feature extraction pipelines efficiently in the joint uneven channel information network, a blend metric loss is proposed. The extra image information is utilized to dynamically adjust the penalty for the sample distance and the distance margin based on the outlier of the hardest samples to construct i-TriHard loss. Besides, softmax loss and center loss are embedded in the proposed loss. The proposed blend metric loss can optimize the spatial distribution and guide the network to learn more discriminative features to enhance the person re-identification performance.
Declarations
Conflict of interest
The authors declare no conflict of interest exists.
Open AccessThis article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/.
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
