CoNet: a lightweight color classification architecture using residual connection and MBConv
摘要
Color classification is a critical task across various industries. Deep learning color classification methods are accurate but have high computational complexity, which limits their use in real-world applications. We presented a pipeline employing a lightweight classification architecture to tackle this issue effectively. The tailored architecture was developed based on the EfficientNet-B0 model, integrating innovative residual connections to improve the extraction of color features. This approach resulted in an exceptionally lightweight model with only 1.90 million parameters and a Giga Floating Point Operations per Second (GFLOPs) count of 0.03 while still delivering outstanding at 94.96% and 78.06% for public vehicle and avocado datasets. We also introduced a new benchmark dataset specifically for Vietnamese motorbike color classification and achieved an accuracy of 91.36%. Our code can be accessed at https://github.com/taipt03/MoCo/tree/main.