KIISE Transactions on Computing Practices (정보과학회 컴퓨팅의 실제 논문지)

Korean Institute of Information Scientists and Engineers (한국정보과학회)
권호 표지
DOI QR코드

DOI QR Code

Mushroom Image Recognition using Convolutional Neural Network and Transfer Learning

컨볼루션 신경망과 전이 학습을 이용한 버섯 영상 인식

  • 강은철 (전북대학교 컴퓨터공학부) ;
  • 한영태 (전북대학교 컴퓨터공학부) ;
  • 오일석 (전북대학교 컴퓨터공학부)
  • Received : 2017.09.06
  • Accepted : 2017.11.21
  • Published : 2018.01.15
⟨ Previous Next ⟩

Abstract

A poisoning accident is often caused by a situation in which people eat poisonous mushrooms because they cannot distinguish between edible mushrooms and poisonous mushrooms. In this paper, we propose an automatic mushroom recognition system by using the convolutional neural network. We collected 1478 mushroom images of 38 species using image crawling, and used the dataset for learning the convolutional neural network. A comparison experiment using AlexNet, VGGNet, and GoogLeNet was performed using the collected datasets, and a comparison experiment using a class number expansion and a fine-tuning technique for transfer learning were performed. As a result of our experiment, we achieve 82.63% top-1 accuracy and 96.84% top-5 accuracy on test set of our dataset.

독버섯 중독 사건이 종종 발생한다. 본 논문은 딥러닝 기술을 활용한 버섯 인식 시스템을 제안한다. 딥러닝 기법 중 하나인 컨볼루션 신경망을 사용하였다. 컨볼루션 신경망을 학습하기 위해 이미지 크롤링을 이용하여 38종의 버섯에 대해 1478장의 영상을 수집하였다. 수집한 데이터셋을 가지고 AlexNet, VGGNet, GoogLeNet을 비교 실험하였으며, 클래스 수 확장에 따른 비교 실험, 전이 학습을 사용한 비교실험을 하였다. 실험 결과 1순위 정확도는 82.63%, 5순위 정확도는 96.84%라는 성능을 얻었다.

Keywords

Acknowledgement

Supported by : 정보 통신 기술 진흥 센터

References

  1. "Be cautious of eating poisonous mushroom when going ancestral graves... 213 people poisoned, 15 people killed in 10 years," Yonhap news agency, 2016.09.15.
  2. Duch-Hyun Cho, "Mushroom of Korea," Deawon Publishing Co., Ltd., 1997.
  3. Il-seok Oh, "Computer Vision," Hanbit Academy, Inc., 2014.
  4. Neeraj Kumar, Peter N. Belhumeur, Arijit Biswas, David W. Jacobs, W. John Kress, Ida C. Lopez, Joao V. B. Soares, "Leafsnap: A computer vision system for automatic plant species identification," European Conference on Computer Vision, Vol. 7573, pp. 502-516, 2012.
  5. Ian Goodfellow, Yoshua Bengio, and Aaron Courville, Deep Learning, MIT Press, 2016.
  6. Alex Krizhevsky, Ilya Sutskever, Geoffrey E. Hinton, "Imagenet classification with deep convolutional neural networks," Advances in neural information processing systems, pp. 1097-1105, 2012.
  7. Karen Simonyan, Andrew Zisserman, "Very deep convolutional networks for large-scale image recognition," arXiv preprint arXiv:1409.1556, 2014.
  8. Christian Szegedy, Wei Liu, Yangqing Jia, Pierre Sermanet, Scott Reed, Dragomir Anguelov, Dumitru Erhan, Vincent Vanhoucke, Andrew Rabinovich, "Going deeper with convolutions," Proc. of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1-9, 2015.
  9. Euncheol Kang, Yeongtae Han, Il-seok Oh, "Mushroom Image Recognition using Convolutional Neural Network," Proc. of Korea Computer Congress 2017, pp. 896-898, 2017.
  10. Yangqing Jia, Evan Shelhamer, Jeff Donahue, Sergey Karayev, Jonathan Long, Ross Girshick, Sergio Guadarrama, Trevor Darrell, "Caffe: Convolutional architecture for fast feature embedding," Proc. of the 22nd ACM international conference on Multimedia, pp. 675-678, 2014.