Upload processor

image_processing_efficientnet.py

@@ -0,0 +1,25 @@
+from transformers.image_processing_utils import BaseImageProcessor
+from configuration_efficientnet import MODEL_NAMES
+from timm import create_model
+from timm.data import resolve_data_config
+from timm.data.transforms_factory import create_transform
+
+class EfficientNetImageProcessor(BaseImageProcessor):
+    model_input_names = ["pixel_values"]
+
+    def __init__(self, 
+                 model_name: str,
+                 **kwargs
+                 ):
+        super().__init__(**kwargs)
+
+        self.config = resolve_data_config({}, model=create_model(model_name, pretrained=False))
+        
+
+    def preprocess(self, image):
+        transforms = create_transform(**self.config)
+        return transforms(image).unsqueeze(0)
+
+__all__ = [
+    "EfficientNetImageProcessor"
+]

preprocessor_config.json

@@ -0,0 +1,26 @@
+{
+  "auto_map": {
+    "AutoImageProcessor": "image_processing_efficientnet.EfficientNetImageProcessor"
+  },
+  "config": {
+    "crop_mode": "center",
+    "crop_pct": 0.875,
+    "input_size": [
+      3,
+      224,
+      224
+    ],
+    "interpolation": "bicubic",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ]
+  },
+  "image_processor_type": "EfficientNetImageProcessor"
+}