优化

.gitignore

 *.pth
+*.png
\ No newline at end of file

app/services/feature_extractor.py

@@ -112,7 +112,11 @@ class FeatureExtractor:
 
         if color:
             mask_img = process_single_image(self.schp_model, img)
-            img = extract_color_region_simple(img, mask_img, color)
+            mask_img.save(r"D:\work\image_search\mask.png")
+            extract_img = extract_color_region_simple(img, mask_img, color)
+            if extract_img:
+                img = extract_img
+
 
         try:
             # 调整图像大小并添加填充

app/services/image_upload.py

@@ -39,7 +39,8 @@ class ImageUpload:
         for image2key in filtered_image2keys:
             image = image2key["image"]
             key = image2key["key"]
-            vector = self.feature_extractor.extract_from_url(image)
+            part = image2key["part"]
+            vector = self.feature_extractor.extract_from_url(image, part)
             images.append(image)
             keys.append(key)
             vectors.append(vector)