diff --git a/app/service/generate_image/utils/image_processing.py b/app/service/generate_image/utils/image_processing.py index ed9fb59..303fc52 100644 --- a/app/service/generate_image/utils/image_processing.py +++ b/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)