Portal de Periódicos da CAPES

An Improved ResNet-50 for Garbage Image Classification

2022; Faculty of Mechanical Engineering in Slavonski Brod, Faculty of Electrical Engineering in Osijek, Faculty of Civil Engineering i; Volume: 29; Issue: 5 Linguagem: Inglês

10.17559/tv-20220420124810

1848-6339

Xiaoxuan Ma, Zhiwen Li, Lei Zhang,

Brain Tumor Detection and Classification

In order to solve the classification model's shortcomings, this study suggests a new trash classification model that is generated by altering the structure of the ResNet-50 network.The improvement is divided into two sections.The first section is to change the residual block.To filter the input features, the attention module is inserted into the residual block.Simultaneously, the downsampling process in the residual block is changed to decrease information loss.The second section is multi-scale feature fusion.To optimize feature usage, horizontal and vertical multi-scale feature fusion is integrated to the primary network structure.Because of the filtering and reuse of image features, the enhanced model can achieve higher classification performance than existing models for small data sets with few samples.The experimental results show that the modified model outperforms the original ResNet-50 model on the TrashNet dataset by 7.62% and is more robust.In the meanwhile, our model is more accurate than other advanced methods.