Sidewalks classification with PyTorch