Upload 3 files
Browse files- image_feature.py +6 -3
- requirements.txt +4 -3
image_feature.py
CHANGED
@@ -146,14 +146,17 @@ def xiang(upload_image):
|
|
146 |
|
147 |
image = upload_image.convert('RGB')
|
148 |
|
149 |
-
processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
|
150 |
-
model = AutoModel.from_pretrained('facebook/dinov2-giant')
|
|
|
|
|
|
|
151 |
|
152 |
inputs = processor(images=image, return_tensors="pt").to(DEVICE)
|
153 |
outputs = model(**inputs)
|
154 |
# last_hidden_states = outputs.last_hidden_state
|
155 |
|
156 |
-
return outputs
|
157 |
|
158 |
except Exception as e:
|
159 |
print(f"发生了一个错误: {e}")
|
|
|
146 |
|
147 |
image = upload_image.convert('RGB')
|
148 |
|
149 |
+
# processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
|
150 |
+
# model = AutoModel.from_pretrained('facebook/dinov2-giant')
|
151 |
+
|
152 |
+
processor = AutoImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
|
153 |
+
model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
|
154 |
|
155 |
inputs = processor(images=image, return_tensors="pt").to(DEVICE)
|
156 |
outputs = model(**inputs)
|
157 |
# last_hidden_states = outputs.last_hidden_state
|
158 |
|
159 |
+
return outputs.last_hidden_state
|
160 |
|
161 |
except Exception as e:
|
162 |
print(f"发生了一个错误: {e}")
|
requirements.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
|
|
|
|
|
|
1 |
transformers==4.39.1
|
2 |
-
|
3 |
-
gradio
|
4 |
-
pillow
|
|
|
1 |
+
numpy==1.26.3
|
2 |
+
Pillow==10.2.0
|
3 |
+
torch==2.1.2
|
4 |
transformers==4.39.1
|
5 |
+
opencv-python
|
|
|
|