Research on Household Waste Detection System Based on Deep Learning

摘要

Household waste is threatening the urban environment increasingly day by day for people’s material needs increasing with the acceleration of urbanization. In this paper, a new waste sorting model is proposed to solve the problems of waste sorting. The style transfer was used to increase the data set to make some objects be sorted well. Then the rotational attention mechanism model was used to increase the accuracy of waste sorting of the blocked objects. The representation vector extraction module in the target tracking algorithm Deep Sort was replaced with Siamese network to make the network more lightweight. As a result, this paper effectively solves the current waste sorting tasks.