creative-graphic-design
/

BASNet

Feature Extraction

Model card Files Files and versions

shunk031 commited on Jul 24, 2024

Commit

c04f6d7

·

verified ·

1 Parent(s): 0895a78

Upload processor

Files changed (1) hide show

image_processing_basnet.py +1 -2

image_processing_basnet.py CHANGED Viewed

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

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