System and method for produce detection and classification

Abstract

Systems, methods, and computer-readable storage media for object detection and classification, and particularly produce detection and classification. A system configured according to this disclosure can receiving, at a processor, an image of an item. The system can then perform, across multiple pre-trained neural networks, feature detection on the image, resulting in feature maps of the image. These feature maps can be concatenated and combined, then input into an additional neural network for feature detection on the combined feature map, resulting in tiered neural network features. The system then classifies, via the processor, the item based on the tiered neural network features.

Description

BACKGROUND

1. Technical Field

The present disclosure relates to object detection, and more specifically to object detection on produce using a combination of multiple classification models.

2. Introduction

Currently, inspecting fruit, vegetables, and other produce for grocery stores requires human beings to manually inspect the produce to verify the quality. For example, as produce is received at a Grocery Distribution Center (GDC), Quality Check (QC) associates inspect the freshness and quality of all produce received, thereby allowing the associates the ability to accept or reject an inbound shipment before it is distributed to the retail locations. This inspection process involves a complete manual inspection executed by the QC associate with the results recorded in a computer system. Each produce category has a set of standardized rules for the quality check, with different types of possible defects which the associate needs to look for and, if the defects are found, which need to be documented.

For example, inspection of strawberries requires (1) selecting and opening of a clamshell (an individual package of strawberries) from a shipped case of multiple strawberry clamshells; (2) counting and recording number of strawberries present in the individual clamshell; (3) inspecting for any defective strawberries; (4) recording the amount and severity of the defects identified; and (5) taking/recording pictures of the defective strawberries as evidence.

During the inspection of produce, an average 50% of the time is spent on counting the produce and defect identification. This manual process is prone to human errors and biased inspection results (depending on the experience, perspective and training of a QC associate). This adds delay in GDC processing time, prolonging the time to reach stores, and thus reduces shelf life of the produce.

Technical Problem

How to combine feature sets of different deep learning architectures used in image processing to enhance object detection and defect classification.

SUMMARY

Additional features and advantages of the disclosure will be set forth in the description which follows, and in part will be obvious from the description, or can be learned by practice of the herein disclosed principles. The features and advantages of the disclosure can be realized and obtained by means of the instruments and combinations particularly pointed out in the appended claims. These and other features of the disclosure will become more fully apparent from the following description and appended claims, or can be learned by the practice of the principles set forth herein.

An exemplary method performed according to the concepts disclosed herein can include: receiving, at a processor, an image of an item; performing, via the processor using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image; concatenating the first feature map, resulting in a first concatenated feature map; performing, via the processor using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image; concatenating the second feature map, resulting in a second concatenated feature map; combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map; performing, via the processor using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; and classifying, via the processor, the item based on the tiered neural network features.

An exemplary system configured according to the concepts disclosed herein can include: a processor; and a computer-readable storage medium having instructions stored which, when executed by the processor, cause the processor to perform operations such as: receiving an image of an item; performing, using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image; concatenating the first feature map, resulting in a first concatenated feature map; performing, using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image; concatenating the second feature map, resulting in a second concatenated feature map; combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map; performing, using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; and classifying the item based on the tiered neural network features.

An exemplary non-transitory computer-readable storage medium configured as disclosed herein can have instructions stored which, when executed by a processor, cause the processor to perform operations which can include: receiving an image of an item; performing, using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image; concatenating the first feature map, resulting in a first concatenated feature map; performing, using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image; concatenating the second feature map, resulting in a second concatenated feature map; combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map; performing, using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; and classifying the item based on the tiered neural network features.

BRIEF DESCRIPTION OF THE DRAWINGS

FIG. 1 illustrates exemplary image recognition using a neural network;

FIG. 2 illustrates a first exemplary concatenation of feature maps from multiple pre-trained networks;

FIG. 3 illustrates a second exemplary concatenation of feature maps from multiple pre-trained networks;

FIG. 4 illustrates an example method claim; and

FIG. 5 illustrates an example computer system.

DETAILED DESCRIPTION

Various embodiments of the disclosure are described in detail below. While specific implementations are described, it should be understood that this is done for illustration purposes only. Other components and configurations may be used without parting from the spirit and scope of the disclosure.

Deep Convolution Neural Networks (CNNs) are the state of the art for classifying images. Many deep learning researchers have come up with a variety of different deep learning architectures like VGG (Visual Geometry Group), Resnet, Inception, etc., which can achieve high rates of accuracy. Despite these high rates of accuracy, the need exists to obtain even higher accuracy within image processing and classification. To obtain that higher accuracy, systems configured according to the principles and concepts disclosed herein leverage feature sets generated by distinct model architectures to achieve state of the art performance on our data.

More specifically, systems configured according to this disclosure use a deep learning architecture which achieves a better performance than the other image evaluation tools that are currently available. This is accomplished by combining features from available pre-trained networks, combining and/or concatenating the features identified by those pre-trained networks, and performing additional analysis on the combined/concatenated features to obtained an output which has a higher accuracy than any single pre-trained network alone.

For example, pre-trained networks such as Inception, Densenet, Xception, etc., provide feature maps for data which is input to those networks. By combining the features of the respective feature maps, we can obtain new features which are complementary to the existing features of the original feature maps. In addition, using convolution and dense layers on the combined feature maps, we can further enhance the features, both the features obtained from the original feature maps and those new features identified based on relationships between features found in distinct, original feature maps. By combining different feature sets for data, and specifically images, where the feature sets are received from different deep learning architectures, category-specific defect classification and object detection are enhanced.

While the disclosed solutions can be applied to any combination of distinct neural network architectures, examples provided herein will be primarily directed to image classification and object detection within images. Implementation of the disclosed concepts and principles, when applied to image classification and object detect, improve the accuracy and efficacy of counting and quality detection systems, which can have real-world benefits. For example, by using the disclosed systems and processes in produce detection, correctly identifying defects within the produce as disclosed herein reduces the manpower required to verify the produce quality, thereby allowing faster movement of produce to stores from distribution centers. This will result in maximizing the shelf life of an item, a reduction in the wasted produce, and providing better-quality produce to customers.

As another example of the utility of these concepts, the improved accuracy in detecting defects within produce can be leveraged to train new Quality Center associates with the help of artificial intelligence. As the system gains information and knowledge about what constitutes a defect, the system can improve how it combines, analyzes, and processes the features from the feature maps. More specifically, as feature maps are combined and/or concatenated, the system then inputs those combined/concatenated feature maps into an additional neural network. As the system improves (through the collection of data and identification of relationships), the system can modify this additional neural network, resulting in a dynamic, changing, and constantly improving system for combining the results of distinct neural networks.

The disclosed solutions take advantage of deep learning and computer vision tools to extract the information from inspection image. The process involves two phases—Object Detection and Object Classification.

Regarding object detection, the implementation disclosed herein can use Faster R-CNN (Regional Convolutional Neural Network), a faster version of object detection than object detection performed using traditional object detection on Convolutional Neural Networks (and identified as “Faster R-CNN” because it is faster than the original application of CNNs to object detection, the application of R-CNNs, and the “fast R-CNN” algorithms developed). The Faster R-CNN implementation disclosed herein can use a Caffe2 framework for object detection: that is, the top-left and bottom-right coordinates of the rectangular regions which might contain objects are discovered. The output of the above object detection algorithm is then fed into the object classification.

The object classification can use an ensemble (more than one) of pre-trained, deep convolutional neural networks along with fine-tuned additional layers for classification. The new, ensemble architecture is then trained using a neural network library (such as Keras (a high-level API (Application Programming Interface) used to build and train deep learning models) with a machine learning framework (such as TensorFlow™)). Preferably, the neural network library selected produces models which are modular, composable, user-friendly, and easy to extend into new layers, loss functions, etc. The multiple CNNs respectively produce models based on the object detected, then the models are trained and updated using the neural network library.

The models can be generated a single time for each respective CNN based on the type of object being detected. Thus for strawberries, multiple models can be produced and trained by the respective CNNs using strawberry object detection, then further trained using neural network library, and further augmented using a machine learning framework.

The multiple models can then be combined and compared, resulting in higher rates of correct categorization of the produce. Over time, the models can continue to be refined and augmented. In addition, the weights or values of the models can be modified based on the accuracy, speed, or efficiency of the respective models. For example, if the ensemble of models produced gives five models, and one of the five models produces false positives thirty percent of the time, and the other four models produce false positives less than 20 percent of the time, the model producing the higher number of false positives can be weighted lower when making the ultimate categorization of the system.

Exemplary produce items on which this detection system and the accompanying ensemble characterization system can be used can include: strawberries, multiple varieties of potatoes, tomatoes, lettuce, etc. The disclosed system has been tested on strawberries, using production data of around 20,000 inspected defective strawberries from GDCs (1 year of data). After exploratory data analysis, fifteen different defects were found in strawberries. The top five defects (decay, bruise, discoloration, overripe soft berries and color) accounted for 96% of the defective strawberries. For the classes which had relatively less data, various image augmentation techniques to augment the data. The models produced used machine learning (ML) with GPU (Graphic Processing Unit) capabilities for model training and for exposing these models as APIs. The APIs can then be integrated into distribution center mobile devices (such as TC70s), so the QC associates performing the quality checks can use their mobile devices to automatically (and systematically) identify defects within objects (such as produce) based on quantifiable standards.

This solution helps to reduce the time taken for the quality inspection in a GDC by up to 70%. The advantages of using such a system is that it provides consistency of inspection without any bias, improves the relationship with the suppliers with standardized inspection process, and speeds up the time for on-shelf delivery. This will also let the QC associates use their time more productively on other tasks, such as movement of produce, ripeness testing, etc. Additionally, store-level produce associates generally have less average experience/training, and this innovation will empower the associates to become quality inspection specialists through deep learning and computer vision capabilities in a very short time.

To prioritize defects within the image processing, one mechanism which can be implemented is a Pareto analysis, where a particular category is defined to capture the defects which occur 80% (or another predefined percentage) of the time. These percentages can vary from model to model, pre-trained network to pre-trained network, within the ensemble of neural networks which initially analyze the data.

While the concepts disclosed herein are focused on using deep learning/computer vision for object detection and classification to aid in quality inspection, and one use of the disclosed invention is for quality control on produce products, the concepts disclosed herein can be leveraged to automate other similar processes in other systems utilizing neural networks.

Turning to the figures, FIG. 1 illustrates exemplary image recognition using a neural network. In this example, a camera 104 takes a picture of a produce product 102, resulting in an image 106. The image 106 is compared to other images stored in an image database 108, and unrelated images are removed 110. The system identifies defects 112 within the image 106 based on the related images, and generates a feature map 114 of the features within the image 106. While inputs and processing capabilities may differ across different neural networks, one or more portions of this process (such as the comparison to the image database 108, removal of unrelated images 110, identification of defects 112, and generation of feature maps 114) may be incorporated into the neural network.

FIG. 2 illustrates a first exemplary concatenation of feature maps from multiple pre-trained networks. In this example there are three pre-trained neural networks 202, 204, 206. Each of these pre-trained neural networks produce a corresponding feature map 208, 210, 212, which are in turn concatenated. These concatenated feature maps 214, 216, 218 are then combined 220. The combined, concatenated feature map 220 is then analyzed/processed to identify additional features 222. These new features 222 were undetected using any individual pre-trained neural network 202, 204, 206, but were detected using the combined results of multiple neural networks. To identify the new features 222, the system can input the combined, concatenated feature map 220 into an additional neural network. This additional neural network can be created based on the specific pre-trained neural networks 202, 204, 206 used in analyzing the initial data.

As an example, an image can be input to multiple pre-trained neural networks 202, 204, 206. Each of those networks 202, 204, 206 produce a respective feature map 208, 210, 212 of the image. The feature maps 208, 210, 212 can identify, for example, objects within the image (such as an apple or strawberry) as well as aspects of those objects (such as a bruise or blemish on fruit). The system concatenates these feature maps (reducing the amount of memory required to store the feature maps to a lower amount) and combines the feature maps together. In some cases, the combination can rely on coordinates built into the feature maps which identify where the various objects and object features are located. These concatenated, combined feature maps are then input to an additional neural network, designed specifically for the pre-trained neural networks 202, 204, 206 initially used to evaluate the image. This additional neural network identifies, based on features within the concatenated, combined feature map, additional features which were missed by each of the original pre-trained neural networks 202, 204, 206. With the features originally identified by the pre-trained neural networks 202, 204, 206, and with the newly identified features identified by the additional neural network, the system can identify and classify the objects within the image. This identification and classification is both more complete and more accurate than any single pre-trained neural networks 202, 204, 206 alone.

FIG. 3 illustrates a second exemplary concatenation of feature maps from multiple pre-trained networks. In this example, there are two pre-trained neural networks 302, 306, which each produce respective results 304, 308 based on the common inputs provided to the neural networks 302, 306. These results 304, 308 are concatenated and combined 310, then those concatenated, combined results are input into an additional neural network 312. From that additional neural network 312, the system produces new results “Result C” 314, which were not found by either of the two initial neural networks 302, 306. The system can then identify and classify the data being evaluated based on the results 304, 308 of the two initial neural networks 302, 306 as well as the additional result 314 of the additional neural network 312.

FIG. 4 illustrates an example method claim. In this example, the system receives receiving, at a processor, an image of an item (402). The system then performs, via the processor using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image (404), and concatenates the first feature map, resulting in a first concatenated feature map (406). The system also performs, via the processor using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image (408), and concatenates the second feature map, resulting in a second concatenated feature map (410). In some configurations, the feature detection using the first and second pre-trained neural networks can occur in parallel, thereby reducing the time required to obtain results. In addition, in some configurations, more than two pre-trained neural networks can be used. The system combines the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map (412), and performs, via the processor using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features (414). In some cases, rather than a third “pre-trained” neural network, the third neural network can be generated upon receiving the feature maps from the first and second pre-trained neural networks, with the third neural network being generated specifically to accommodate for known differences between the first pre-trained neural network and the second pre-trained neural network. The system then classifies, via the processor, the item based on the tiered neural network features.

In some configurations, the item can be produce. In such cases, the feature detection can identify defects within the produce.

In some configurations, at least one of the first pre-trained neural network, the second pre-trained neural network, and the third pre-trained neural network is a Faster Regional Convolutional Neural Network. In such cases, the Faster Regional Convolutional Neural Network identifies a top-left coordinate of a rectangular region for each item within the image and a bottom-right coordinate of the rectangular region.

In some configurations, the third pre-trained neural network uses distinct neural links (connections between the nodes of the neural network) than the neural links of the first pre-trained neural network and the second pre-trained neural network.

In some configurations, the processor is a Graphical Processing Unit, rather than a generic processor.

With reference to FIG. 5, an exemplary system includes a general-purpose computing device 500, including a processing unit (CPU or processor) 520 and a system bus 510 that couples various system components including the system memory 530 such as read-only memory (ROM) 540 and random access memory (RAM) 550 to the processor 520. The system 500 can include a cache of high-speed memory connected directly with, in close proximity to, or integrated as part of the processor 520. The system 500 copies data from the memory 530 and/or the storage device 560 to the cache for quick access by the processor 520. In this way, the cache provides a performance boost that avoids processor 520 delays while waiting for data. These and other modules can control or be configured to control the processor 520 to perform various actions. Other system memory 530 may be available for use as well. The memory 530 can include multiple different types of memory with different performance characteristics. It can be appreciated that the disclosure may operate on a computing device 500 with more than one processor 520 or on a group or cluster of computing devices networked together to provide greater processing capability. The processor 520 can include any general purpose processor and a hardware module or software module, such as module 1562, module 2564, and module 3566 stored in storage device 560, configured to control the processor 520 as well as a special-purpose processor where software instructions are incorporated into the actual processor design. The processor 520 may essentially be a completely self-contained computing system, containing multiple cores or processors, a bus, memory controller, cache, etc. A multi-core processor may be symmetric or asymmetric.

The system bus 510 may be any of several types of bus structures including a memory bus or memory controller, a peripheral bus, and a local bus using any of a variety of bus architectures. A basic input/output (BIOS) stored in ROM 540 or the like, may provide the basic routine that helps to transfer information between elements within the computing device 500, such as during start-up. The computing device 500 further includes storage devices 560 such as a hard disk drive, a magnetic disk drive, an optical disk drive, tape drive or the like. The storage device 560 can include software modules 562, 564, 566 for controlling the processor 520. Other hardware or software modules are contemplated. The storage device 560 is connected to the system bus 510 by a drive interface. The drives and the associated computer-readable storage media provide nonvolatile storage of computer-readable instructions, data structures, program modules and other data for the computing device 500. In one aspect, a hardware module that performs a particular function includes the software component stored in a tangible computer-readable storage medium in connection with the necessary hardware components, such as the processor 520, bus 510, display 570, and so forth, to carry out the function. In another aspect, the system can use a processor and computer-readable storage medium to store instructions which, when executed by the processor, cause the processor to perform a method or other specific actions. The basic components and appropriate variations are contemplated depending on the type of device, such as whether the device 500 is a small, handheld computing device, a desktop computer, or a computer server.

Although the exemplary embodiment described herein employs the hard disk 560, other types of computer-readable media which can store data that are accessible by a computer, such as magnetic cassettes, flash memory cards, digital versatile disks, cartridges, random access memories (RAMs) 550, and read-only memory (ROM) 540, may also be used in the exemplary operating environment. Tangible computer-readable storage media, computer-readable storage devices, or computer-readable memory devices, expressly exclude media such as transitory waves, energy, carrier signals, electromagnetic waves, and signals per se.

To enable user interaction with the computing device 500, an input device 590 represents any number of input mechanisms, such as a microphone for speech, a touch-sensitive screen for gesture or graphical input, keyboard, mouse, motion input, speech and so forth. An output device 570 can also be one or more of a number of output mechanisms known to those of skill in the art. In some instances, multimodal systems enable a user to provide multiple types of input to communicate with the computing device 500. The communications interface 580 generally governs and manages the user input and system output. There is no restriction on operating on any particular hardware arrangement and therefore the basic features here may easily be substituted for improved hardware or firmware arrangements as they are developed.

The steps outlined herein are exemplary and can be implemented in any combination thereof, including combinations that exclude, add, or modify certain steps.

Use of language such as “at least one of X, Y, and Z” or “at least one or more of X, Y, or Z” are intended to convey a single item (just X, or just Y, or just Z) or multiple items (i.e., {X and Y}, {Y and Z}, or {X, Y, and Z}). “At least one of” is not intended to convey a requirement that each possible item must be present.

The various embodiments described above are provided by way of illustration only and should not be construed to limit the scope of the disclosure. Various modifications and changes may be made to the principles described herein without following the example embodiments and applications illustrated and described herein, and without departing from the spirit and scope of the disclosure.

Claims

1. A method comprising: receiving, at a processor, an image of an item;performing, via the processor using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image;concatenating the first feature map, resulting in a first concatenated feature map;performing, via the processor using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image;concatenating the second feature map, resulting in a second concatenated feature map;combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map;performing, via the processor using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; andclassifying, via the processor, the item based on the tiered neural network features, the classifying including implementing a set of pre-trained neural networks, the set of pre-trained neural networks having been produced based on the tiered neural network features, the classification being a combination of results of the set of pre-trained neural networks, and a result of each pre-trained neural network of the set of pre-trained neural networks being weighted based on a corresponding accuracy.

2. The method of claim 1, wherein the item is produce.

3. The method of claim 2, wherein the feature detection identifies defects within the produce.

4. The method of claim 1, wherein at least one of the first pre-trained neural network, the second pre-trained neural network, and the third pre-trained neural network is a Faster Regional Convolutional Neural Network.

5. The method of claim 4, wherein the Faster Regional Convolutional Neural Network identifies a top-left coordinate of a rectangular region for each item within the image and a bottom-right coordinate of the rectangular region.

6. The method of claim 1, wherein the third pre-trained neural network uses distinct neural links than the neural links of the first pre-trained neural network and the second pre-trained neural network.

7. The method of claim 1, wherein the processor is a Graphical Processing Unit.

8. A system, comprising: a processor; and a computer-readable storage medium having instructions stored which, when executed by the processor, cause the processor to perform operations comprising: receiving an image of an item;performing, using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image;concatenating the first feature map, resulting in a first concatenated feature map;performing, using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image;concatenating the second feature map, resulting in a second concatenated feature map;combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map;performing, using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; andclassifying the item based on the tiered neural network features, the classifying including implementing a set of pre-trained neural networks, the set of pre-trained neural networks having been produced based on the tiered neural network features, the classification being a combination of results of the set of pre-trained neural networks, and a result of each pre-trained neural network of the set of pre-trained neural networks being weighted based on a corresponding accuracy.

9. The system of claim 8, wherein the item is produce.

10. The system of claim 9, wherein the feature detection identifies defects within the produce.

11. The system of claim 8, wherein at least one of the first pre-trained neural network, the second pre-trained neural network, and the third pre-trained neural network is a Faster Regional Convolutional Neural Network.

12. The system of claim 11, wherein the Faster Regional Convolutional Neural Network identifies a top-left coordinate of a rectangular region for each item within the image and a bottom-right coordinate of the rectangular region.

13. The system of claim 8, wherein the third pre-trained neural network uses distinct neural links than the neural links of the first pre-trained neural network and the second pre-trained neural network.

14. The system of claim 8, wherein the processor is a Graphical Processing Unit.

15. A non-transitory computer-readable storage medium having instructions stored which, when executed by a computing device, cause the computing device to perform operations comprising: receiving an image of an item;performing, using a first pre-trained neural network, feature detection on the image, resulting in a first feature map of the image;concatenating the first feature map, resulting in a first concatenated feature map;performing, using a second pre-trained neural network, feature detection on the image, resulting in a second feature map of the image;concatenating the second feature map, resulting in a second concatenated feature map;combining the first concatenated feature map and the second concatenated feature map, resulting in a combined feature map;performing, using a third pre-trained neural network, feature detection on the combined feature map, resulting in tiered neural network features; andclassifying the item based on the tiered neural network features, the classifying including implementing a set of pre-trained neural networks, the set of pre-trained neural networks having been produced based on the tiered neural network features, the classification being a combination of results of the set of pre-trained neural networks, and a result of each pre-trained neural network of the set of pre-trained neural networks being weighted based on a corresponding accuracy.

16. The non-transitory computer-readable storage medium of claim 15, wherein the item is produce.

17. The non-transitory computer-readable storage medium of claim 16, wherein the feature detection identifies defects within the produce.

18. The non-transitory computer-readable storage medium of claim 15, wherein at least one of the first pre-trained neural network, the second pre-trained neural network, and the third pre-trained neural network is a Faster Regional Convolutional Neural Network.

19. The non-transitory computer-readable storage medium of claim 18, wherein the Faster Regional Convolutional Neural Network identifies a top-left coordinate of a rectangular region for each item within the image and a bottom-right coordinate of the rectangular region.

20. The non-transitory computer-readable storage medium of claim 15, wherein the third pre-trained neural network uses distinct neural links than the neural links of the first pre-trained neural network and the second pre-trained neural network.