change mask colors
Browse files
main.py
CHANGED
|
@@ -65,9 +65,9 @@ async def predict_image(file: UploadFile = File(...)):
|
|
| 65 |
background_area = 0
|
| 66 |
|
| 67 |
mask_dict = {
|
| 68 |
-
'acne': (
|
| 69 |
-
'dry': (
|
| 70 |
-
'oily': (
|
| 71 |
'background': (255,255,255)
|
| 72 |
}
|
| 73 |
# change the mask to base64 and calculate the score
|
|
|
|
| 65 |
background_area = 0
|
| 66 |
|
| 67 |
mask_dict = {
|
| 68 |
+
'acne': (245, 177, 177),
|
| 69 |
+
'dry': (208, 181, 166),
|
| 70 |
+
'oily': (240, 230, 214),
|
| 71 |
'background': (255,255,255)
|
| 72 |
}
|
| 73 |
# change the mask to base64 and calculate the score
|