Python Image Classification Code For Excavation