Upload 2 files

app.py

@@ -3,7 +3,8 @@ import image_feature as func
 
 
 def work11(image1, image2):
- return func.infer1(image1, image2)
+ return func.xiang(image1)
+ # return func.infer1(image1, image2)
  # return func.infer4(image1, image2)
 
 

image_feature.py

@@ -140,4 +140,26 @@ def infer1(image1, image2):
  # 无论是否发生异常，都会执行此代码块
  print("这是finally块")
 
+# 输出图片向量
+def xiang(url):
+ try:
+
+ image = Image.open(url).convert('RGB')
+
+ processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
+ model = AutoModel.from_pretrained('facebook/dinov2-giant')
+
+ inputs = processor(images=image, return_tensors="pt").to(DEVICE)
+ outputs = model(**inputs)
+ # last_hidden_states = outputs.last_hidden_state
+
+ return outputs
+
+ except Exception as e:
+ print(f"发生了一个错误: {e}")
+ return '异常'+ str(e)
+ finally:
+ # 无论是否发生异常，都会执行此代码块
+ print("这是finally块")
+