creative-graphic-design
/

ISNet-general-use

Feature Extraction

Model card Files Files and versions Community

shunk031 commited on Jul 24, 2024

Commit

0d3cd14

·

verified ·

1 Parent(s): 16b0e01

Upload processor

Files changed (1) hide show

image_processing_isnet.py +1 -2

image_processing_isnet.py CHANGED Viewed

@@ -6,8 +6,7 @@ from PIL import Image
 from PIL.Image import Image as PilImage
 from torchvision import transforms
 from torchvision.transforms.functional import normalize
-from transformers.image_processing_base import BatchFeature
-from transformers.image_processing_utils import BaseImageProcessor
 from transformers.image_utils import ImageInput

 from PIL.Image import Image as PilImage
 from torchvision import transforms
 from torchvision.transforms.functional import normalize
+from transformers.image_processing_utils import BaseImageProcessor, BatchFeature
 from transformers.image_utils import ImageInput