Abstract
In efforts to build a hairstyle recommendation application, we are confronted with the lack of relevant hairstyle-related datasets. In this paper, we present a new large-scale dataset for hairstyle recommendation, CelebHair, based on the celebrity facial attributes dataset, CelebA. Our dataset inherited the majority of facial images along with some beauty-related facial attributes from CelebA. Additionally, we employed facial landmark detection techniques to extract extra features such as nose length and pupillary distance, and deep convolutional neural networks for face shape and hairstyle classification. Empirical comparison has demonstrated the superiority of our dataset to other existing hairstyle-related datasets regarding variety, veracity, and volume. Analysis and experiments have been conducted on the dataset in order to evaluate its robustness and usability.
Y. Chen and Y. Zhang—Co-first authors with equal contribution.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Liu, Z., Luo, P., Wang, X., Tang, X.: Deep learning face attributes in the wild. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 3730–3738 (2015)
King, D.E.: Dlib-ml: a machine learning toolkit. J. Mach. Learn. Res. 10, 1755–1758 (2009)
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934 (2020)
Lama, N.: Face shape dataset (2020). https://www.kaggle.com/niten19/face-shape-dataset
LeCun, Y., Haffner, P., Bottou, L., Bengio, Y.: Object recognition with gradient-based learning. Shape, Contour and Grouping in Computer Vision. LNCS, vol. 1681, pp. 319–345. Springer, Heidelberg (1999). https://doi.org/10.1007/3-540-46805-6_19
Jaderberg, M., Simonyan, K., Zisserman, A., Kavukcuoglu, K.: Spatial transformer networks. arXiv preprint arXiv:1506.02025 (2015)
Yin, W., Fu, Y., Ma, Y., Jiang, Y.G., Xiang, T., Xue, X.: Learning to generate and edit hairstyles. In: Proceedings of the 25th ACM International Conference on Multimedia, pp. 1627–1635 (2017)
Breiman, L.: Machine learning. Random Forest. 45(1), 5–32 (2001)
Liu, L., Xing, J., Liu, S., Xu, H., Zhou, X., Yan, S.: Wow! you are so beautiful today!. ACM Tran. Multimedia Comput. Commun. Appl. (TOMM) 11(1s), 1–22 (2014)
Weerasinghe, H., Vidanagama, D.: Machine learning approach for hairstyle recommendation. In: 2020 5th International Conference on Information Technology Research (ICITR), pp. 1–4. IEEE (2020)
Sunhem, W., Pasupa, K.: An approach to face shape classification for hairstyle recommendation. In: 2016 Eighth International Conference on Advanced Computational Intelligence (ICACI), pp. 390–394. IEEE (2016)
Pasupa, K., Sunhem, W., Loo, C.K.: A hybrid approach to building face shape classifier for hairstyle recommender system. Expert Syst. Appl. 120, 14–32 (2019)
Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. In: International conference on machine learning, pp. 448–456. PMLR (2015)
Maas, A.L., Hannun, A.Y., Ng, A.Y.: Rectifier nonlinearities improve neural network acoustic models. In: Proceedings of the International Conference on Machine Learning (ICML), vol. 30, p. 3. Citeseer (2013)
Bradski, G.: The opencv library. Dr Dobb’s J. Softw. Tools 25, 120–125 (2000)
Niu, Z., Zhou, M., Wang, L., Gao, X., Hua, G.: Ordinal regression with multiple output cnn for age estimation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4920–4928 (2016)
Eidinger, E., Enbar, R., Hassner, T.: Age and gender estimation of unfiltered faces. IEEE Trans. Inf. Forensics Secur. 9(12), 2170–2179 (2014)
Zhang, Z., Song, Y., Qi, H.: Age progression/regression by conditional adversarial autoencoder. In: IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE (2017)
Wang, W., Chen, Y., Kao, C.: Integrating augmented reality into female hairstyle try-on experience. In: 2011 Seventh International Conference on Natural Computation, vol. 4, pp. 2125–2127 (2011). https://doi.org/10.1109/ICNC.2011.6022427
Earl, M.: Switching eds: Face swapping with python, dlib, and opencv (2015). https://matthewearl.github.io/2015/07/28/switching-eds-with-python/
Acknowledgments
This work is supported by Research Innovation Fund for College Students of Beijing University of Posts and Telecommunications, and National Natural Science Foundation of China under Grant No. 61702043.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Chen, Y., Zhang, Y., Huang, Z., Luo, Z., Chen, J. (2021). CelebHair: A New Large-Scale Dataset for Hairstyle Recommendation Based on CelebA. In: Qiu, H., Zhang, C., Fei, Z., Qiu, M., Kung, SY. (eds) Knowledge Science, Engineering and Management. KSEM 2021. Lecture Notes in Computer Science(), vol 12817. Springer, Cham. https://doi.org/10.1007/978-3-030-82153-1_27
Download citation
DOI: https://doi.org/10.1007/978-3-030-82153-1_27
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-82152-4
Online ISBN: 978-3-030-82153-1
eBook Packages: Computer ScienceComputer Science (R0)