Flower Image Classification System Based on Lightweight DCNN
CSTR:
Author:
Affiliation:

1.College of Computer and Information Science, Three Gorges University, Yichang 443002, China;2.Hubei Key Laboratory of Intelligent Vision Based Monitoring for Hydroelectric Engineering (Three Gorges University), Yichang 443002, China;3.Information Center, Sinopharm Gezhouba Central Hospital, Yichang 443002, China

Clc Number:

TP391.41;TP183

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    To solve the problem that deep convolutional neural network (DCNN) models with heavy weights are difficult to be effectively applied on AI edge devices with weak computing power and high storage costs, a flower image classification system equipped with a lightweight DCNN is proposed with the help of a heavyweight DCNN during training process. First, an extended flower data set suitable for lightweight DCNN training is constructed by using a heavyweight DCNN combined with transferring learning, the crawler technology and the maximum connected region segmentation method. Then, two lightweight DCNN models, Tiny-Darknet and Darknet-Reference, oriented for devices with weak computer power are trained based on the specially built flower image gallery. Experimental results show that the two optimized models obtained can achieve 98.07% and 98.83% average classification accuracy respectively on Oxford102 flower dataset while keeping the model size as 4 MB and 28 MB, which have promising application potentials for AI edge computer devices.

    Reference
    Related
    Cited by
Get Citation

XU Guangzhu, ZHU Zequn, YIN Silu, LIU Gaofei, Lei Bangjun. Flower Image Classification System Based on Lightweight DCNN[J].,2021,36(4):756-768.

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:August 28,2020
  • Revised:March 09,2021
  • Adopted:
  • Online: July 25,2021
  • Published:
Article QR Code