AI-based object classification