Waste-Multi-Class-Classfication-using-PyTorch