feat generate sketch 污点检测改为顶部两角

app/service/generate_image/utils/image_processing.py

@@ -205,3 +205,8 @@ def generate_category_recognition(image, gender):
         else:
             category = 'Outwear'
     return category, scores, image
+
+
+if __name__ == '__main__':
+    image = cv2.imread("1.png")
+    remove_background(image)