1. Designing optimal object detection networks for detecting damages to canola kernels
- Author
-
Mann, Danny (Biosystems Engineering), Major, Arkady (Electrical and Computer Engineering), Erkinbaev, Chyngyz, Thakuria, Angshuman, Mann, Danny (Biosystems Engineering), Major, Arkady (Electrical and Computer Engineering), Erkinbaev, Chyngyz, and Thakuria, Angshuman
- Abstract
Canola is an essential Canadian crop that generated a revenue of $14.4 B from its export in the 2022 fiscal year. To consistently export the best quality canola and set the correct pricing that truly reflects the grade, it is crucial to invest in reliable, high speed, and accurate grading technologies. Leveraging the current progress in Artificial Intelligence algorithms, this research proposes a comprehensive end-to-end system to detect damage to canola kernels and grade them. The proposed system comprises an accelerated sample preparation setup, custom-built specialized AI models, and an edge-AI microprocessor for in-field use. This thesis mainly focuses on the development of specialized neural networks to accurately detect damaged canola seeds as it is the most complex part of the system. Two foundational object detection networks, You Only Look Once (YOLO) version 5 and version 7 were optimized to be compatible with resource-constrained hardware environments. The aim of developing the two optimal networks was to mitigate trade-offs among speed, accuracy, cost, and model size, thereby creating a more balanced and optimized network. Several architectural design options were explored to compress the network structure of the two models in terms of size and cost yet retain or improve the metrics of accuracy and inference speed. The results from the design choices indicate that reconstructing YOLOv5 with ShuffleNet as the backbone reduces its size and cost and increases the inference speed but negatively affects the detection accuracy. Replacing the Convolutional Blocks present in the Spatial Pyramid Pooling Cross Stage Partial and all the Efficient Layer Aggregation Network modules of YOLOv7 with the Ghost Convolutional Network and adding two Convolutional Block Attention Modules improves both mean average precision metric compared to the parent model and reduces the size and cost. To prepare the samples faster, a semi-automatic option was adopted and its effect on d
- Published
- 2024