End-to-End Face Parsing via Interlinked Convolutional Neural Networks
This work addresses a specific bottleneck in face parsing for computer vision applications, offering an incremental improvement over existing methods.
The paper tackles the problem of limited performance in two-stage face parsing models by introducing an end-to-end framework called STN-iCNN, which integrates a Spatial Transformer Network to enable joint training and improves accuracy, achieving competitive results on the Helen Dataset and superior performance on the CelebAMask-HQ dataset.
Face parsing is an important computer vision task that requires accurate pixel segmentation of facial parts (such as eyes, nose, mouth, etc.), providing a basis for further face analysis, modification, and other applications. Interlinked Convolutional Neural Networks (iCNN) was proved to be an effective two-stage model for face parsing. However, the original iCNN was trained separately in two stages, limiting its performance. To solve this problem, we introduce a simple, end-to-end face parsing framework: STN-aided iCNN(STN-iCNN), which extends the iCNN by adding a Spatial Transformer Network (STN) between the two isolated stages. The STN-iCNN uses the STN to provide a trainable connection to the original two-stage iCNN pipeline, making end-to-end joint training possible. Moreover, as a by-product, STN also provides more precise cropped parts than the original cropper. Due to these two advantages, our approach significantly improves the accuracy of the original model. Our model achieved competitive performance on the Helen Dataset, the standard face parsing dataset. It also achieved superior performance on CelebAMask-HQ dataset, proving its good generalization. Our code has been released at https://github.com/aod321/STN-iCNN.