This research presents an improved artificial intelligence system for automatically grading Stropharia rugoso-annulata (wine cap) mushrooms based on their size and quality. The new system uses advanced computer vision techniques to analyze mushroom images in real-time, achieving 95.2% accuracy while being efficient enough to run on smaller computing devices used in food processing facilities. By combining wavelet analysis for capturing overall mushroom shape with a streamlined attention mechanism, the system successfully grades mushrooms faster and more consistently than manual sorting, potentially reducing labor costs in industrial mushroom production.