Upload 2 files

image_processing_tagger.py

@@ -1,11 +1,8 @@
-# copied from ViTImageProcessor (https://github.com/huggingface/transformers/blob/v4.37.2/src/transformers/models/vit/image_processing_vit.py)
-
-"""Image processor class for WD v14 Tagger."""
+"""Image processor class for WD Tagger."""
 
 from typing import Optional, List, Dict, Union, Tuple
 
 import numpy as np
-import cv2
 from PIL import Image
 
 from transformers.image_processing_utils import (
@@ -36,7 +33,6 @@ from transformers.utils import TensorType, logging
 logger = logging.get_logger(__name__)
 
 
-# copied from transformers.image_transforms.resize
 def resize_with_padding(
     image: np.ndarray,
     size: Tuple[int, int],
@@ -140,9 +136,9 @@ def resize_with_padding(
     return new_image
 
 
-class WDv14TaggerImageProcessor(BaseImageProcessor):
+class WDTaggerImageProcessor(BaseImageProcessor):
     r"""
-    Constructs a WD v14 Tagger image processor.
+    Constructs a WD Tagger image processor.
 
     Args:
         do_resize (`bool`, *optional*, defaults to `True`):

preprocessor_config.json

@@ -1,7 +1,7 @@
 {
-  "image_processor_type": "WDv14TaggerImageProcessor",
+  "image_processor_type": "WDTaggerImageProcessor",
   "auto_map": {
-    "AutoImageProcessor": "image_processing_tagger.WDv14TaggerImageProcessor"
+    "AutoImageProcessor": "image_processing_tagger.WDTaggerImageProcessor"
   },
   "color": [
     255,