from .policy import ImageClassificationPolicy